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

Fix GatherND to output dynamic rank tensor (#3073)

Browse Source 
Signed-off-by: Roman Kazantsev <roman.kazantsev@intel.com>
This commit is contained in:
Roman Kazantsev 2020-11-11 14:47:00 +03:00 committed by GitHub
parent b20f76967a
commit 53911a91f6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ngraph/core/src/op/gather_nd.cpp
View File

@ -142,7 +142,7 @@ void op::v5::GatherND::validate_and_infer_types()
}
else
{
set_output_type(0, data_type, PartialShape{Dimension::dynamic()});
set_output_type(0, data_type, PartialShape::dynamic());
}
}

12
ngraph/test/type_prop/gather_nd.cpp
View File

@ -96,6 +96,18 @@ TEST(type_prop, gather_nd_batch_dim2_with_dyn_dim3)
ASSERT_TRUE(G5->get_output_partial_shape(0).same_scheme(out_shape));
}
TEST(type_prop, gather_nd_batch_dim0_with_dyn_ind_dim)
{
PartialShape params_shape{
7, Dimension::dynamic(), Dimension::dynamic(), 12, Dimension::dynamic()};
PartialShape indices_shape{7, 5, 3, Dimension::dynamic()};
auto P = make_shared<op::Parameter>(element::f32, params_shape);
auto I = make_shared<op::Parameter>(element::i32, indices_shape);
auto G5 = make_shared<op::v5::GatherND>(P, I, 0);
ASSERT_EQ(G5->get_element_type(), element::f32);
ASSERT_TRUE(G5->get_output_partial_shape(0).same_scheme(PartialShape::dynamic()));
}
TEST(type_prop, gather_nd_fail_batch_dims_greater_indices_rank)
{
Shape params_shape{2, 3, 4, 5};