Updated ConcatOptimization to support Concat with 0D input of one dimension (#2012)
authorEvgeny Lazarev <evgeny.lazarev@intel.com>
Wed, 2 Sep 2020 07:21:23 +0000 (10:21 +0300)
committerGitHub <noreply@github.com>
Wed, 2 Sep 2020 07:21:23 +0000 (10:21 +0300)
commitc7bcbb576cccd7ed417506a8002137c6ddf58ae8
tree55532f8ff0299ab1c96f5dc9d5a2cfb4883e1e91
parent315274ad125247206eba26e58cc855200ea5f5e9
Updated ConcatOptimization to support Concat with 0D input of one dimension (#2012)

* Updated ConcatOptimization transformation to work when one dimension of input to Concat is 0D

* Fixed ConcatOptimization transformation to reconnect input edges to Concat

* Completely re-written ConcatOptimization

* Updated Concat0D optimization transformation

* Fixed order of traversing Concat input ports

* Refactored ConcatOptimization transformation to use `delete_input_port` function

* Detele trailing unconnected ports in the ConcatOptimization.py

* Cleaner implementation of ConcatOptimization + unit test
model-optimizer/extensions/middle/ConcatOptimization.py
model-optimizer/extensions/middle/ConcatOptimization_test.py
model-optimizer/mo/graph/graph.py