Add dynamic shape checks to nGraph transformations (#2735)
authorGleb Kazantaev <gleb.kazantaev@intel.com>
Fri, 23 Oct 2020 12:39:47 +0000 (15:39 +0300)
committerGitHub <noreply@github.com>
Fri, 23 Oct 2020 12:39:47 +0000 (15:39 +0300)
commitc4e0b74fb1ebaa3ffee7afb3900f935a300514ee
tree4113ef6a7ca8cbd908376b2d9bd0513fb7d6194c
parent8c97127aa7b795f7e70746725c379b9f2c7b0ffd
Add dynamic shape checks to nGraph transformations (#2735)

* Added dynamic shape checks for BatchNormDecompositoin pass

* Added dynamic shapes checks for FQTranspose fusion pass

* Added patter::has_static_rank predicate

* Added dynamic shapes checks for BroadcastToTiles pass

* Fixed BN inputs order

* Add dynamic shape checks for DepthToSpace/SpaceToDepth passes

* Added dynamic check for ReduceToPooling pass

* Updated BN transformation

* Fix PR comments

* size_t to int64_t

* Updated reduce to pooling pattern
13 files changed:
inference-engine/src/transformations/include/transformations/op_conversions/convert_reduce_to_pooling.hpp
inference-engine/src/transformations/src/transformations/common_optimizations/pull_transpose_through_fq.cpp
inference-engine/src/transformations/src/transformations/op_conversions/batch_norm_decomposition.cpp
inference-engine/src/transformations/src/transformations/op_conversions/convert_broadcast_to_tiles.cpp
inference-engine/src/transformations/src/transformations/op_conversions/convert_depth_to_space.cpp
inference-engine/src/transformations/src/transformations/op_conversions/convert_space_to_depth.cpp
inference-engine/tests/functional/inference_engine/transformations/batch_norm_decompositoin.cpp [new file with mode: 0644]
inference-engine/tests/functional/inference_engine/transformations/convert_broadcast_to_tiles_test.cpp [new file with mode: 0644]
inference-engine/tests/functional/inference_engine/transformations/convert_reduce_to_pooling_test.cpp
inference-engine/tests/functional/inference_engine/transformations/ngraph_depth_to_space_transform_test.cpp
inference-engine/tests/functional/inference_engine/transformations/ngraph_fq_transpose_test.cpp
ngraph/core/include/ngraph/pattern/op/pattern.hpp
ngraph/core/src/pattern/op/pattern.cpp