19 #include "ngraph/op/op.hpp"
51 virtual std::shared_ptr<Node>
52 clone_with_new_inputs(
const OutputVector& new_args)
const override;
55 virtual bool evaluate(
const HostTensorVector& output_values,
56 const HostTensorVector& input_values)
const override;
60 void set_axis(int64_t axis) { m_axis = axis; }
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: one_hot.hpp:28
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
virtual bool evaluate(const HostTensorVector &output_values, const HostTensorVector &input_values) const override
Evaluates the op on input_values putting results in output_values.
OneHot()=default
Constructs a one-hot operation.
OneHot(const Output< Node > &indices, const Output< Node > &depth, const Output< Node > &on_value, const Output< Node > &off_value, int64_t axis)
Constructs a one-hot operation.
int64_t get_axis() const
Definition: one_hot.hpp:59
const NodeTypeInfo & get_type_info() const override
Definition: one_hot.hpp:31
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28