add default 0 value for batch_dims in v1::Gather reference (openvinot…

…oolkit#116) * add default 0 value for batch_dims in v1::Gather reference * finally run on ARM: added static_cast * changed argument types from size_t -> int64_t * changed back to size_t
eshoguli · Apr 12, 2021 · a21aef9 · a21aef9
1 parent 0fd22e3
commit a21aef9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/modules/arm_plugin/src/arm_converter/arm_converter_gather.cpp b/modules/arm_plugin/src/arm_converter/arm_converter_gather.cpp
@@ -17,7 +17,8 @@ template <> Converter::Conversion::Ptr Converter::Convert(const ngraph::op::v1::
                                     node.get_input_shape(0),
                                     node.get_input_shape(1),
                                     node.get_output_shape(0),
-                                    static_cast<size_t>(node.get_axis()));
+                                    static_cast<size_t>(node.get_axis()),
+                                    static_cast<size_t>(0));
     };
 
     return CallSwitch(