IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
Files
38022c4cd636ca0e551396c5e19292893f71f26a
openvino/model-optimizer/extensions/ops
History
Eugeny Volosenkov 38022c4cd6 Mo implementation for If with tf extractor (#6662) 
* Add tf2.x impl for If

* Fix ir_engine

* Fix opset

* Fix BOM file

* Added new test

* Fix comments

* Add subgraph_utils

* Fix comments

* Fix transform

* code refactoring

* Fix description

* rewrite support for empty tensor in if

* added onnx extractor

* delete onnx_if

* fix bug with fake_outputs

* Fix test

* Fix control_flow and fix commentaries

* create method results_mapping_and_finding_fake_outputs(output_nodes_in_subgraph,
2021-08-19 10:13:21 +03:00
..
__init__.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
accum.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
activation_ops.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
adaptive_avg_pooling.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
argmax.py
Implementation of ArgMin ONNX + TF extractors (#5126)
2021-05-06 13:41:49 +03:00
argmin.py
Implementation of ArgMin ONNX + TF extractors (#5126)
2021-05-06 13:41:49 +03:00
assert_op.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
aten.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
axpy.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
BatchNormInference.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
binarization.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
BlockLSTM.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
BN.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
box_nms.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
bucketize.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
Cast.py
numpy.packbits usage for numpy 1.14.0 support (#5078)
2021-04-01 20:11:12 +03:00
constant_fill.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
ConvertLike.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
copyop.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
correlation.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
ctc_greedy_decoder_seq_len.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
ctc_greedy_decoder.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
ctc_loss.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
cumsum.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
data_augmentation.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
depth_to_space.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
dequantize_linear.py
Add axis support (#5242)
2021-05-17 15:03:22 +03:00
detectionoutput_onnx.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
DetectionOutput.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
dft.py
MO support for operations DFT and IDFT (#5197)
2021-05-07 09:44:24 +03:00
einsum.py
Correct layout for Einsum inputs and output (#6696)
2021-07-23 11:01:06 +03:00
elementwise.py
[MO] fix MarkNodesWithShapeValues (#5808)
2021-06-04 18:49:49 +03:00
embedding_bag.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
Enter.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
Exit.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
ExtractImagePatches.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
fake_output.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
fakequantize.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
gather.py
[MO][nG] add Gather-8 to MO, layer tests, nG python api unit-tests and up/down-grading transformations (#6560)
2021-08-13 11:56:49 +03:00
gatherelements.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
gathernd.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
GatherTree.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
gelu.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
grn.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
GRU.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
GRUCell.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
hard_sigmoid.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
identity.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
If.py
Mo implementation for If with tf extractor (#6662)
2021-08-19 10:13:21 +03:00
instance_normalization.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
interp.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
interpolate.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
LookupTableInsert.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
loop.py
Implement transformation for TensorFlow 2 Map Function (aka tf.map_fn) (#6836)
2021-07-30 12:10:59 +03:00
lstm_cell.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
lstm_sequence.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
LSTM.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
MatMul.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
merge.py
[MO] Fixes in Merge infer function (#6125)
2021-06-24 18:53:13 +03:00
mvn.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
mxfft.py
MO support for operations DFT and IDFT (#5197)
2021-05-07 09:44:24 +03:00
mxrepeat.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
mxreshape.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
NextIteration.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
non_max_suppression.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
non_zero.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
normalize_l2.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
normalize.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
one_hot.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
ONNXResize10.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
ONNXResize11.py
Add support resize with 2 inputs (#5927)
2021-06-08 13:15:54 +03:00
pack.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
parameter.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
pnorm.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
power_file.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
prediction_heatmap.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
prelu.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
priorbox_clustered.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
priorbox.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
priorgridgenerator_onnx.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
proposal_onnx.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
proposal_python_example.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
proposal.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
psroipooling.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
quantize_linear.py
Add axis support (#5242)
2021-05-17 15:03:22 +03:00
range.py
[MO] Range output_type correction for FP16 (#6590)
2021-08-02 11:41:12 +03:00
rank.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
ReduceOps.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
regionyolo.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
reorgyolo.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
resize_factor_utils.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
resize.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
reverse_sequence.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
Reverse.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
RNN.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
RNNCell.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
roialign.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
roifeatureextractor_onnx.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
roll.py
MO support for operations DFT and IDFT (#5197)
2021-05-07 09:44:24 +03:00
scatter.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
scatternd.py
Fix ScatterND validation and implement evaluate (#4905)
2021-03-31 19:01:46 +03:00
select.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
shufflechannel.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
simplernms.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
size.py
Added extractor for ONNX operation Size (#6031)
2021-06-04 17:18:01 +03:00
slice_like.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
space_to_depth.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
sparse_fill_empty_rows.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
sparse_reshape.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
sparse_segment_mean.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
sparse_segment_sqrtn.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
sparse_segment_sum.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
spatial_transformer.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
splice.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
split.py
Fix errors for VariadicSplit layer restored from serialized IR (#5613)
2021-05-19 19:03:58 +03:00
stop_gradient.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
swapaxis.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
switch.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
tensor_iterator.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
TensorArray.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
TensorArrayGather.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
TensorArrayRead.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
TensorArrayScatter.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
TensorArraySize.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
TensorArrayWrite.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
TensorIterator_ops.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
TFFFT.py
MO support for operations DFT and IDFT (#5197)
2021-05-07 09:44:24 +03:00
TFResize.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
topk.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
topkrois_onnx.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
transpose.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
unique.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00
upsample.py
Align copyright notice in python scripts (CVS-51320) (#4974)
2021-03-26 17:54:28 +03:00