DepthToSpace, SpaceToDepth layers optimizations (#706)
authorGorokhov Dmitriy <dmitry.gorokhov@intel.com>
Thu, 4 Jun 2020 11:25:19 +0000 (14:25 +0300)
committerGitHub <noreply@github.com>
Thu, 4 Jun 2020 11:25:19 +0000 (14:25 +0300)
commit3183c116d9db1bc87f5184e3bc61b4366a7e0fca
tree02a5f2ab7bbd4170dd51804531f9759256926f82
parent01e60d057da7e64c1fcf255e62a259f34a5f7ba0
DepthToSpace, SpaceToDepth layers optimizations (#706)

* [CPU] Updated DepthToSpace and SpaceToDepth layers to be conformant with the specification

The patch also includes n[d]hwc layout support as well as some optimizations

* [CPU][TESTS] Removed old DepthToSpace test since it doesn't corresponds to layer's specification

* [nGraph] Utilize CommonOptimizations pass with custom transformations callback
18 files changed:
inference-engine/src/cldnn_engine/cldnn_engine.cpp
inference-engine/src/legacy_api/src/ie_layer_validators.cpp
inference-engine/src/mkldnn_plugin/mkldnn_plugin.cpp
inference-engine/src/mkldnn_plugin/nodes/depth_to_space.cpp
inference-engine/src/mkldnn_plugin/nodes/space_to_depth.cpp
inference-engine/src/transformations/include/transformations/common_optimizations/common_optimizations.hpp
inference-engine/src/transformations/src/transformations/common_optimizations/common_optimizations.cpp
inference-engine/src/transformations/src/transformations/depth_to_space_fusion.cpp
inference-engine/tests/functional/plugin/cpu/shared_tests_instances/single_layer_tests/depth_to_space.cpp [new file with mode: 0644]
inference-engine/tests/functional/plugin/cpu/shared_tests_instances/single_layer_tests/space_to_depth.cpp [new file with mode: 0644]
inference-engine/tests/functional/plugin/shared/include/single_layer_tests/depth_to_space.hpp [new file with mode: 0644]
inference-engine/tests/functional/plugin/shared/include/single_layer_tests/space_to_depth.hpp [new file with mode: 0644]
inference-engine/tests/functional/plugin/shared/src/single_layer_tests/depth_to_space.cpp [new file with mode: 0644]
inference-engine/tests/functional/plugin/shared/src/single_layer_tests/space_to_depth.cpp [new file with mode: 0644]
inference-engine/tests/ngraph_functions/include/ngraph_functions/builders.hpp
inference-engine/tests/ngraph_functions/src/depth_to_space.cpp [new file with mode: 0644]
inference-engine/tests/ngraph_functions/src/space_to_depth.cpp [new file with mode: 0644]
inference-engine/tests_deprecated/unit/engines/mkldnn/graph/layers/extensions/depth_to_space_tests.cpp [deleted file]