openvino/model-optimizer/mo/front
Roman Kazantsev 032cebb9da
Implement transformation for TensorFlow 2 Map Function (aka tf.map_fn) (#6836)
* Implement transformation for TensorFlow 2 Map Function primitive

Signed-off-by: Roman Kazantsev <roman.kazantsev@intel.com>

* Add a description for get_external_node_by_internal_id function

Signed-off-by: Roman Kazantsev <roman.kazantsev@intel.com>

* Correct a name for get_external_nodes_by_internal_id function

Signed-off-by: Roman Kazantsev <roman.kazantsev@intel.com>

* Fix a description for get_external_nodes_by_internal_id function

Signed-off-by: Roman Kazantsev <roman.kazantsev@intel.com>

* Add logging and fix indentation

Signed-off-by: Roman Kazantsev <roman.kazantsev@intel.com>

* Use skip_nodes_by_condition to by-pass StopGradient nodes for tf.map_fn

Signed-off-by: Roman Kazantsev <roman.kazantsev@intel.com>
2021-07-30 12:10:59 +03:00
..
caffe Error during Caffe model conversion with Python3.8 (#6056) 2021-06-08 16:09:04 +03:00
common Moved MO unit test files to a separate directory (#5312) 2021-04-20 14:47:18 +03:00
kaldi Kaldi priors (#6258) 2021-07-02 15:18:23 +05:00
mxnet Additional telemetry events in MO. (#5662) 2021-05-24 15:21:29 +03:00
onnx Additional telemetry events in MO. (#5662) 2021-05-24 15:21:29 +03:00
tf Implement transformation for TensorFlow 2 Map Function (aka tf.map_fn) (#6836) 2021-07-30 12:10:59 +03:00
__init__.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00
extractor.py [MO] Add support to moc_frontend of ":" as delimiter for --input and --output (#6543) 2021-07-29 13:09:51 +03:00
subgraph_matcher.py Align copyright notice in python scripts (CVS-51320) (#4974) 2021-03-26 17:54:28 +03:00