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

[IE][VPU][nGraph]: Fixes Reshape's shape infer method (#1327)

Browse Source 
Previously, if Reshape had input pattern with values [0, -1] - it
propagated dynamic shape through a function. At the same time,
taking "0" and "-1" interpretation into consideration, it turns out
in such cases we could just propagate the same input shape in case of
2D input.

For Faster-RCNN this fix makes static dimensions on dynamic paths static.

Signed-off-by: Gladilov, Gleb <gleb.gladilov@intel.com>
This commit is contained in:
Gladilov, Gleb 2020-07-15 22:17:36 +03:00 committed by GitHub
parent 2803498995
commit a0d60abef7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
ngraph/src/ngraph/op/reshape.cpp
View File

@ -338,6 +338,14 @@ void op::v1::Reshape::validate_and_infer_types()
}
}
}
if (out_shape_val == std::vector<std::int64_t>{0, -1} &&
input_pshape.rank().is_static() && input_pshape.rank().get_length() == 2)
{
partial_shape[0] = input_pshape[0];
partial_shape[1] = input_pshape[1];
}
set_output_type(0, get_input_element_type(0), PartialShape(partial_shape));
}
}