IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix unintended shape change (#4510)

Browse Source
This commit is contained in:
Svetlana Dolinina 2021-02-26 18:29:29 +03:00 committed by GitHub
parent b0043bb599
commit a1ef9498a3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model-optimizer/extensions/ops/loop.py
View File

@ -119,7 +119,7 @@ class Loop(TensorIterator):
loop_port_idx = record['external_port_id']
if loop_port_idx != -1: # the id = -1 for execution condition output which is not connected anywhere
output_value = body_node.in_port(0).data.get_value()
output_shape = body_node.in_port(0).data.get_shape()
output_shape = body_node.in_port(0).data.get_shape().copy()
concat_axis = record['axis']
if concat_axis is not None:
assert output_shape[concat_axis] == 1, 'Dimension for concatenation is not equal to 1 for scan ' \