Commit 663e9bb4 authored by Ilya Churaev's avatar Ilya Churaev Committed by Sang Ik Lee

Fixed creation NonMaxSuppression op with dynamic shapes (#4119)

* Fixed creation NonMaxSuppression op with dynamic shapes

* Fixed code style
Co-authored-by: 's avatarRobert Kimball <robert.kimball@intel.com>
Co-authored-by: 's avatarScott Cyphers <diyessi@users.noreply.github.com>
Co-authored-by: 's avatarMichał Karzyński <postrational@users.noreply.github.com>
Co-authored-by: 's avatarSang Ik Lee <sang.ik.lee@intel.com>
parent d38e160a
......@@ -69,6 +69,11 @@ void op::v1::NonMaxSuppression::validate_and_infer_types()
{
const auto boxes_ps = get_input_partial_shape(0);
const auto scores_ps = get_input_partial_shape(1);
if (boxes_ps.is_dynamic() || scores_ps.is_dynamic())
{
set_output_type(0, get_input_element_type(0), PartialShape::dynamic(Rank::dynamic()));
return;
}
NODE_VALIDATION_CHECK(this,
boxes_ps.rank().is_static() && static_cast<size_t>(boxes_ps.rank()) == 3,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment