19 #include "ngraph/axis_vector.hpp"
20 #include "ngraph/node.hpp"
21 #include "ngraph/op/op.hpp"
22 #include "ngraph/runtime/aligned_buffer.hpp"
23 #include "ngraph/runtime/host_tensor.hpp"
34 static constexpr
NodeTypeInfo type_info{
"ScatterElementsUpdate", 3};
51 virtual std::shared_ptr<Node>
52 clone_with_new_inputs(
const OutputVector& inputs)
const override;
54 const HostTensorVector& inputs)
const override;
57 bool evaluate_scatter_element_update(
const HostTensorVector& outputs,
58 const HostTensorVector& inputs)
const;
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
Definition: scatter_elements_update.hpp:32
virtual void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
const NodeTypeInfo & get_type_info() const override
Definition: scatter_elements_update.hpp:35
ScatterElementsUpdate(const Output< Node > &data, const Output< Node > &indices, const Output< Node > &updates, const Output< Node > &axis)
Constructs a ScatterElementsUpdate node.
bool evaluate(const HostTensorVector &outputs, const HostTensorVector &inputs) const override
Evaluates the op on input_values putting results in output_values.
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28