# Remove disabled ops
list(REMOVE_ITEM LIBRARY_SRC
${CMAKE_CURRENT_SOURCE_DIR}/src/op/conv_integer.cpp
- ${CMAKE_CURRENT_SOURCE_DIR}/src/op/gather_nd.cpp
${CMAKE_CURRENT_SOURCE_DIR}/src/op/quant_conv.cpp
)
list(REMOVE_ITEM PUBLIC_HEADERS
${CMAKE_CURRENT_SOURCE_DIR}/include/onnx_import/op/conv_integer.hpp
- ${CMAKE_CURRENT_SOURCE_DIR}/include/onnx_import/op/gather_nd.hpp
${CMAKE_CURRENT_SOURCE_DIR}/include/onnx_import/op/quant_conv.hpp
)
// Disabled in CMakeList
// Update to higher opset required
-#include "ngraph/opsets/opset0.hpp"
+#include "onnx_import/default_opset.hpp"
#include "onnx_import/utils/common.hpp"
namespace ngraph
{
OutputVector gather_nd(const Node& node)
{
- OutputVector ng_inputs{node.get_ng_inputs()};
- auto data = ng_inputs.at(0);
- auto indices = ng_inputs.at(1);
+ const OutputVector ng_inputs{node.get_ng_inputs()};
+ const auto data = ng_inputs.at(0);
+ const auto indices = ng_inputs.at(1);
+ const auto batch_dims = node.get_attribute_value<int64_t>("batch_dims", 0);
- return {std::make_shared<ngraph::opset0::GatherND>(data, indices)};
+ return {std::make_shared<default_opset::GatherND>(data, indices, batch_dims)};
}
} // namespace set_1
#include "onnx_import/op/flatten.hpp"
#include "onnx_import/op/floor.hpp"
#include "onnx_import/op/gather.hpp"
-// #include "onnx_import/op/gather_nd.hpp"
+#include "onnx_import/op/gather_nd.hpp"
#include "onnx_import/op/gemm.hpp"
#include "onnx_import/op/global_average_pool.hpp"
#include "onnx_import/op/global_max_pool.hpp"
REGISTER_OPERATOR("Flatten", 1, flatten);
REGISTER_OPERATOR("Floor", 1, floor);
REGISTER_OPERATOR("Gather", 1, gather);
- // REGISTER_OPERATOR("GatherND", 1, gather_nd);
+ REGISTER_OPERATOR("GatherND", 1, gather_nd);
REGISTER_OPERATOR("Gemm", 1, gemm);
REGISTER_OPERATOR("Gemm", 6, gemm);
REGISTER_OPERATOR("GlobalAveragePool", 1, global_average_pool);
onnx_model_qlinear_matmul
onnx_model_qlinear_matmul_3d
-# Not supported ONNX op: GatherND
+# The indices input type i64 is not supported by the CPU plugin
onnx_model_gatherND_int32
onnx_model_gatherND_float