19 #include "ngraph/op/op.hpp"
32 NGRAPH_RTTI_DECLARATION;
43 const size_t batch_dims = 0);
47 virtual std::shared_ptr<Node>
48 clone_with_new_inputs(
const OutputVector& new_args)
const override;
50 size_t get_batch_dims()
const {
return m_batch_dims; }
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:71
A handle for one of a node's outputs.
Definition: node_output.hpp:42
Root of all actual ops.
Definition: op.hpp:29
GatherND operation.
Definition: gather_nd.hpp:30
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
GatherND(const Output< Node > &data, const Output< Node > &indices, const size_t batch_dims=0)
Constructs a GatherND operation.
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28