openvino/model-optimizer/mo/front/common
Roman Kazantsev 10b18a00c6
Improve reshapeability of models with eltwise nodes influencing shapes (#2767)
* Fix ElementwiseInputReshape transformation

Reshape node always needs to be inserted
in order to preserve ShapeOf nodes (reshapability of a model) that can potentially be above
elementwise node.

Refactor EltwiseInputReshape_test and EltwiseInputNormalization_test since the logic of maintaining reshape for eltwise has been changed.

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

* Merge EltwiseInputNormalization and EltwiseInputReshape transformations

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

* Remove Unsqueeze from Fused_op

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

* Fix code after code review #1

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

* Fix code after review #2

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

* Fix code review #4

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

* Perform full normalization based on shapes of all inputs to eltwise

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

* Refactor much to avoid old API and edges with unsqueeze_dims attribute

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

* Fix code after review

Signed-off-by: Roman Kazantsev <roman.kazantsev@intel.com>
2020-11-16 09:50:41 +03:00
..
extractors Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
partial_infer Improve reshapeability of models with eltwise nodes influencing shapes (#2767) 2020-11-16 09:50:41 +03:00
__init__.py Publishing R3 2018-10-16 13:45:03 +03:00
custom_replacement_registry.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
find_unsupported_ops.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
layout.py publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b 2020-04-15 19:01:57 +03:00
layout_test.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
register_custom_ops.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00
replacement.py publish master branch snapshot, revision cdcab9d7ab48ffb0ee5629fabbfa06cb45debd9b 2020-04-15 19:01:57 +03:00
weights.py Publishing 2020.1 content 2020-02-11 22:48:49 +03:00