openvino/model-optimizer/unit_tests/utils
Pavel Esir d9d63d0d7e
[MO] fix MarkNodesWithShapeValues (#5808)
* fix marking of nodes with shape values: corrected 'leaky' starting condition

* cleaned up code

* fixed t2t type collision error: removed strict assert from Pow; now type_infer is same as for other eltwise ops; added warnings about type alignment

* type cast for Pow according to ONNX spec

* fix highlight in dump_graph_for_graphviz

* soft_get('op')

* Revert "soft_get('op')"

This reverts commit cadfe18f

* Revert "type cast for Pow according to ONNX spec"

This reverts commit bf85ebf8

* applied review comments

* comment for ShapeOf bfs, removed Error from elementwise op

* reverted back dump_graph_for_graphviz: fill_color -> node_attrs

* applied last review comments: returned back Error, added type_infer into unit-test utils
2021-06-04 18:49:49 +03:00
..
test_data Moved MO unit test files to a separate directory (#5312) 2021-04-20 14:47:18 +03:00
__init__.py Moved MO unit test files to a separate directory (#5312) 2021-04-20 14:47:18 +03:00
extractors.py Moved MO unit test files to a separate directory (#5312) 2021-04-20 14:47:18 +03:00
graph.py [MO] fix MarkNodesWithShapeValues (#5808) 2021-06-04 18:49:49 +03:00