IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
Files
5ab628d776d783fbae9ae857f5153823f8dea19f
openvino/model-optimizer/extensions/ops
History
Roman Kazantsev 5ab628d776 Fix TF2 Keras RNN support (#4116) 
It adds support for Bidirectional RNNs and avoids use of ONNX primitives from MO-TF.

Signed-off-by: Roman Kazantsev <roman.kazantsev@intel.com>
2021-02-02 16:00:11 +03:00
..
__init__.py
Publishing R3
2018-10-16 13:45:03 +03:00
accum_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
accum.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
activation_ops.py
ONNX Loop operation support (#2756)
2020-10-27 23:04:43 +03:00
activation_test.py
Fix Mish and SoftPlus value propagation functions (#2120)
2020-09-11 12:58:14 +03:00
adaptive_avg_pooling.py
Update operation attributes (#3814)
2021-01-29 10:08:06 +03:00
argmax_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
argmax.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
assert_op.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
assert_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
aten.py
Implement support for opset3 EmbeddingBag ops (#546)
2020-06-08 18:06:40 +03:00
axpy.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
BatchNormInference.py
Re-implement onnx old-style extractors with extractor extensions (#3459)
2020-12-10 09:24:24 +03:00
binarization.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
BlockLSTM.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
BN.py
Re-implement caffe old-style extractors with extractor extensions (#3675)
2021-02-01 13:17:17 +03:00
box_nms.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
bucketize_test.py
Implement Bucketize in MO and MKLDNN for opset3 (#583)
2020-05-28 11:11:07 +03:00
bucketize.py
Update operation attributes (#3814)
2021-01-29 10:08:06 +03:00
Cast.py
MO clean from IR v7 and other legacy code (#1521)
2020-07-29 17:43:12 +03:00
constant_fill.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
copyop.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
correlation_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
correlation.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
ctc_greedy_decoder_test.py
Extend MO for operation CTCLoss and partly refactor CTCGreedyDecoder (#588)
2020-08-17 19:19:59 +03:00
ctc_greedy_decoder.py
Update operation attributes (#3814)
2021-01-29 10:08:06 +03:00
ctc_loss_test.py
Extend MO for operation CTCLoss and partly refactor CTCGreedyDecoder (#588)
2020-08-17 19:19:59 +03:00
ctc_loss.py
Update operation attributes (#3814)
2021-01-29 10:08:06 +03:00
cumsum_test.py
publish master branch snapshot, revision 8d31237e2c3f673cbb0f0ba110fc10f5cce1d2bb
2020-05-22 02:23:12 +03:00
cumsum.py
Update operation attributes (#3814)
2021-01-29 10:08:06 +03:00
data_augmentation_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
data_augmentation.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
depth_to_space_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
depth_to_space.py
MO clean from IR v7 and other legacy code (#1521)
2020-07-29 17:43:12 +03:00
dequantize_linear.py
Add ONNX DequantizeLinear to MO (#1250)
2020-07-24 18:39:09 +03:00
detectionoutput_onnx.py
Implement ExperimentalDetectronDetectionOutput and ExperimentalDetectronPriorGridGenerator operations as nGraph ops (#3374)
2021-01-18 12:16:40 +03:00
DetectionOutput.py
Update operation attributes (#3814)
2021-01-29 10:08:06 +03:00
div_value_propagation_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
elementwise_test.py
Add Round-5 operation (#2328)
2020-10-20 18:36:19 +03:00
elementwise.py
Actualize operations attributes (#3613)
2020-12-18 11:47:41 +03:00
embedding_bag_test.py
Implement support for opset3 EmbeddingBag ops (#546)
2020-06-08 18:06:40 +03:00
embedding_bag.py
Implement support for opset3 EmbeddingBag ops (#546)
2020-06-08 18:06:40 +03:00
Enter.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
Exit.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
ExtractImagePatches_test.py
ExtractImagePatches MO implementation and nGraph transformation (#739)
2020-08-19 13:23:41 +03:00
ExtractImagePatches.py
ExtractImagePatches MO implementation and nGraph transformation (#739)
2020-08-19 13:23:41 +03:00
fake_output.py
Move FakeOutput resolving to back phase (#2033)
2020-09-07 10:20:24 +03:00
fakequantize.py
Actualize operations attributes (#3613)
2020-12-18 11:47:41 +03:00
gather_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
gather.py
[ MO ] Complete weights layout permutation (#2299)
2020-09-18 14:42:16 +03:00
gatherelements_test.py
nGraph 'shell' implementation for GatherElements-6 and MO 'shell' implementation (#3467)
2020-12-21 15:45:15 +03:00
gatherelements.py
nGraph 'shell' implementation for GatherElements-6 and MO 'shell' implementation (#3467)
2020-12-21 15:45:15 +03:00
gathernd_test.py
Extend MO for operation GatherND (#2540)
2020-10-15 07:40:58 +03:00
gathernd.py
Extend MO for operation GatherND (#2540)
2020-10-15 07:40:58 +03:00
GatherTree.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
gelu.py
Publishing 2020.2 content
2020-04-13 21:17:23 +03:00
grn_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
grn.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
GRU.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
GRUCell.py
Update operation attributes (#3814)
2021-01-29 10:08:06 +03:00
hard_sigmoid.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
identity.py
[ MO TF ] IdentityN support (#529)
2020-05-25 10:52:58 +03:00
instance_normalization_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
instance_normalization.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
interp.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
interpolate_test.py
Extend MO for support of Interpolate-4 (#2026)
2020-09-09 16:28:52 +03:00
interpolate.py
Update operation attributes (#3814)
2021-01-29 10:08:06 +03:00
LookupTableInsert_test.py
Implement LookupTableInsert shape inference (#2348)
2020-10-20 09:57:55 +03:00
LookupTableInsert.py
Implement LookupTableInsert shape inference (#2348)
2020-10-20 09:57:55 +03:00
loop.py
Fix TF2 Keras RNN support (#4116)
2021-02-02 16:00:11 +03:00
lstm_cell.py
LSTMCell/Sequence v1, reference implementations and decompose transformations for LSTM/GRU/RNN Cells (#2000)
2020-09-04 09:04:36 +03:00
lstm_sequence.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
LSTM.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
MatMul_test.py
[MO] Aligned MatMul-1 with spec (#3337)
2020-12-01 13:02:14 +03:00
MatMul_value_propagation_test.py
[MO] Aligned MatMul-1 with spec (#3337)
2020-12-01 13:02:14 +03:00
MatMul.py
Update operation attributes (#3814)
2021-01-29 10:08:06 +03:00
merge_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
merge.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
mvn.py
Update operation attributes (#3814)
2021-01-29 10:08:06 +03:00
mxrepeat.py
Publishing 2020.2 content
2020-04-13 21:17:23 +03:00
mxreshape.py
Publishing 2020.2 content
2020-04-13 21:17:23 +03:00
NextIteration.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
non_max_suppression_test.py
NMS-4 op support (#1115)
2020-06-30 14:04:31 +03:00
non_max_suppression.py
Update operation attributes (#3814)
2021-01-29 10:08:06 +03:00
non_zero.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
normalize_l2.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
normalize_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
normalize.py
MO clean from IR v7 and other legacy code (#1521)
2020-07-29 17:43:12 +03:00
one_hot_test.py
Leaky relu transformation refactor (#2640)
2020-10-14 16:43:29 +03:00
one_hot.py
MO clean from IR v7 and other legacy code (#1521)
2020-07-29 17:43:12 +03:00
ONNXResize11_test.py
retinanet onnx model has Upsample error (#3052)
2020-11-11 17:31:00 +03:00
ONNXResize11.py
retinanet onnx model has Upsample error (#3052)
2020-11-11 17:31:00 +03:00
pack.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
parameter.py
MO clean from IR v7 and other legacy code (#1521)
2020-07-29 17:43:12 +03:00
pnorm.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
power_file.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
prediction_heatmap.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
prelu.py
MO clean from IR v7 and other legacy code (#1521)
2020-07-29 17:43:12 +03:00
priorbox_clustered_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
priorbox_clustered.py
Update operation attributes (#3814)
2021-01-29 10:08:06 +03:00
priorbox_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
priorbox.py
Update operation attributes (#3814)
2021-01-29 10:08:06 +03:00
priorgridgenerator_onnx.py
Implement ExperimentalDetectronDetectionOutput and ExperimentalDetectronPriorGridGenerator operations as nGraph ops (#3374)
2021-01-18 12:16:40 +03:00
proposal_onnx.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
proposal_python_example.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
proposal_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
proposal.py
Update operation attributes (#3814)
2021-01-29 10:08:06 +03:00
psroipooling_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
psroipooling.py
Update operation attributes (#3814)
2021-01-29 10:08:06 +03:00
quantize_linear.py
[MO] Support ONNX QuantizeLinear (#1451)
2020-08-10 21:10:45 +03:00
quantize_test.py
Publishing 2020.2 content
2020-04-13 21:17:23 +03:00
range.py
Add Range-4 support (#1552)
2020-08-19 07:58:41 +03:00
rank.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
ReduceOps_test.py
Enable ReduceL1 and ReduceL2 operations (#1799)
2020-08-17 16:32:53 +03:00
ReduceOps.py
Update operation attributes (#3814)
2021-01-29 10:08:06 +03:00
regionyolo_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
regionyolo.py
Update operation attributes (#3814)
2021-01-29 10:08:06 +03:00
reorgyolo_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
reorgyolo.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
resample.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
resize_factor_utils.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
resize.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
reverse_sequence.py
publish master branch snapshot, revision 59af1853ca21ea08acf17b177da0b239753deb46
2020-05-18 17:21:58 +03:00
Reverse_test.py
Publishing 2020.2 content
2020-04-13 21:17:23 +03:00
Reverse.py
Publishing 2020.2 content
2020-04-13 21:17:23 +03:00
RNN.py
Publishing 2020.2 content
2020-04-13 21:17:23 +03:00
RNNCell.py
publish master branch snapshot, revision 59af1853ca21ea08acf17b177da0b239753deb46
2020-05-18 17:21:58 +03:00
roialign.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
roifeatureextractor_onnx.py
Implement ExperimentalDetectronROIFeatureExtractor operation as nGraph op (#3739)
2021-01-20 16:20:54 +03:00
scatter.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
scatternd_test.py
Implement ScatterND operation in MO and transform for SparseToDense (#584)
2020-07-10 12:29:15 +03:00
scatternd.py
Implement ScatterND operation in MO and transform for SparseToDense (#584)
2020-07-10 12:29:15 +03:00
select_test.py
Fix unit tests for select layer. (#638)
2020-06-08 18:39:40 +03:00
select.py
Actualize operations attributes (#3613)
2020-12-18 11:47:41 +03:00
shufflechannel.py
publish master branch snapshot, revision 8d31237e2c3f673cbb0f0ba110fc10f5cce1d2bb
2020-05-22 02:23:12 +03:00
simplernms_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
simplernms.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
size.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
slice_like_test.py
[MO MXNET] Fixed spatial reshape on GluonCV models (#587)
2020-06-30 22:32:13 +03:00
slice_like.py
[MO MXNET] Fixed spatial reshape on GluonCV models (#587)
2020-06-30 22:32:13 +03:00
space_to_depth_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
space_to_depth.py
MO clean from IR v7 and other legacy code (#1521)
2020-07-29 17:43:12 +03:00
sparse_fill_empty_rows_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
sparse_fill_empty_rows.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
sparse_reshape_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
sparse_reshape.py
[MO] Fix many spelling and grammar errors (#2543)
2020-10-09 12:16:12 +03:00
sparse_segment_mean_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
sparse_segment_mean.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
sparse_segment_sqrtn_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
sparse_segment_sqrtn.py
[MO] Fix many spelling and grammar errors (#2543)
2020-10-09 12:16:12 +03:00
sparse_segment_sum_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
sparse_segment_sum.py
[MO] Fix many spelling and grammar errors (#2543)
2020-10-09 12:16:12 +03:00
spatial_transformer_test.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
spatial_transformer.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
splice.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
split_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
split.py
[MO] Fix many spelling and grammar errors (#2543)
2020-10-09 12:16:12 +03:00
stop_gradient.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
swapaxis.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
switch_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
switch.py
Publishing 2020.1 content
2020-02-11 22:48:49 +03:00
tensor_iterator.py
ONNX Loop operation support (#2756)
2020-10-27 23:04:43 +03:00
TensorArray.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
TensorArrayGather.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
TensorArrayRead.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
TensorArrayScatter.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
TensorArraySize.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
TensorArrayWrite.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
TensorIterator_ops.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
topk_test.py
MO clean from IR v7 and other legacy code (#1521)
2020-07-29 17:43:12 +03:00
topk.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
topkrois_onnx.py
publish master branch snapshot, revision 49482ae3bea0cbaa07474f86f36db11943142687
2020-05-13 21:12:22 +03:00
transpose_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
transpose.py
[ MO ] Complete weights layout permutation (#2299)
2020-09-18 14:42:16 +03:00
unique_test.py
publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b
2020-04-15 19:01:57 +03:00
unique.py
[MO] Fix many spelling and grammar errors (#2543)
2020-10-09 12:16:12 +03:00
upsample_test.py
publish master branch snapshot, revision 0110d9c98fd7209589d06344f0d836f61d81f4b3
2020-04-27 21:21:29 +03:00
upsample.py
Improve support ONNX Resize-10 created by PyTorch (#1350)
2020-10-29 15:26:23 +03:00