data_rank check for axis

openvinotoolkit · Dec 17, 2020 · 8b798f6 · 8b798f6
1 parent 59d5c23
commit 8b798f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ngraph/core/src/op/gather_elements.cpp b/ngraph/core/src/op/gather_elements.cpp
@@ -50,7 +50,7 @@ void op::v6::GatherElements::validate_and_infer_types()
     auto indices_rank = indices_pshape.rank();
 
     int64_t axis = m_axis;
-    if (m_axis < 0)
+    if (m_axis < 0 && data_rank.is_static())
         axis += data_rank.get_length();
 
     set_output_type(0, data_type, indices_pshape);