19 #include "ngraph/op/op.hpp"
33 static constexpr NodeTypeInfo type_info{
"ScatterBase", 3};
34 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
50 const Output<Node>& indices,
51 const Output<Node>& updates,
52 const Output<Node>& axis);
56 static constexpr
int DATA = 0;
57 static constexpr
int INDICES = 1;
58 static constexpr
int UPDATES = 2;
59 static constexpr
int AXIS = 3;
virtual void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:70
Base class for ScatterXXX operators.
Definition: scatter_base.hpp:31
ScatterBase(const Output< Node > &inputs, const Output< Node > &indices, const Output< Node > &updates, const Output< Node > &axis)
Constructs ScatterBase object.
const NodeTypeInfo & get_type_info() const override
Definition: scatter_base.hpp:34
Root of all actual ops.
Definition: op.hpp:29