19 #include "ngraph/op/op.hpp"
48 NGRAPH_RTTI_DECLARATION;
68 virtual std::shared_ptr<Node>
69 clone_with_new_inputs(
const OutputVector& new_args)
const override;
76 m_auto_broadcast = auto_broadcast;
80 virtual bool evaluate(
const HostTensorVector& output_values,
81 const HostTensorVector& input_values)
const override;
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
Elementwise selection operation.
Definition: select.hpp:46
Select(const Output< Node > &arg0, const Output< Node > &arg1, const Output< Node > &arg2, const AutoBroadcastSpec &auto_broadcast=AutoBroadcastSpec(AutoBroadcastType::NUMPY))
Constructs a selection operation.
virtual bool evaluate(const HostTensorVector &output_values, const HostTensorVector &input_values) const override
Evaluates the op on input_values putting results in output_values.
const AutoBroadcastSpec & get_autob() const override
Definition: select.hpp:79
Select()
Constructs a selection operation.
Definition: select.hpp:50
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
AutoBroadcastType
Specifies the algorithm to use for implicit broadcasting of a tensor to align with another tensor.
Definition: attr_types.hpp:167
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28
Implicit broadcast specification.
Definition: attr_types.hpp:323