From a21aef9668efd34a72f892c50d01ea5b5d921d54 Mon Sep 17 00:00:00 2001 From: Pavel Esir Date: Mon, 12 Apr 2021 17:01:29 +0300 Subject: [PATCH] add default 0 value for batch_dims in v1::Gather reference (#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 --- modules/arm_plugin/src/arm_converter/arm_converter_gather.cpp | 3 ++- 1 file changed, 2 insertions(+), 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 index 263e8bc9f209ca..3fa8aba8108b6c 100644 --- 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(node.get_axis())); + static_cast(node.get_axis()), + static_cast(0)); }; return CallSwitch(