openvino/model-optimizer/mo
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
..
back [MO] Fix many spelling and grammar errors (#2543) 2020-10-09 12:16:12 +03:00
front Improve reshapeability of models with eltwise nodes influencing shapes (#2767) 2020-11-16 09:50:41 +03:00
graph Fix the NHWC->NCHW transformation for dynamic weights (#2848) 2020-11-06 19:04:46 +03:00
middle Leaky relu transformation refactor (#2640) 2020-10-14 16:43:29 +03:00
ops Added transformation config to support automl efficientdet models (#2894) 2020-11-02 19:21:05 +03:00
pipeline [MO] Fix many spelling and grammar errors (#2543) 2020-10-09 12:16:12 +03:00
utils 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
bom_test.py Publishing 2020.2 content 2020-04-13 21:17:23 +03:00
main.py [ MO ] KSO=ON for Kaldi (#2028) 2020-10-23 13:14:00 +03:00
main_test.py Cleanup IR v7 from the MO (#1008) 2020-06-22 11:52:00 +03:00