19 #include "ngraph/op/op.hpp"
38 virtual std::shared_ptr<Node>
39 clone_with_new_inputs(
const OutputVector& new_args)
const override;
43 element::Type get_output_type()
const {
return m_output_type; }
44 void set_output_type(
element::Type output_type) { m_output_type = output_type; }
46 using Node::set_output_type;
54 void set_is_foldable(
bool is_foldable) { m_is_foldable = is_foldable; }
55 bool get_is_foldable()
const {
return m_is_foldable; }
56 bool evaluate(
const HostTensorVector& output_values,
57 const HostTensorVector& input_values)
const override;
58 bool evaluate_lower(
const HostTensorVector& output_values)
const override;
59 bool evaluate_upper(
const HostTensorVector& output_values)
const override;
60 bool constant_fold(OutputVector& output_values,
61 const OutputVector& input_values)
override;
64 bool m_is_foldable =
true;
75 NGRAPH_RTTI_DECLARATION;
81 virtual std::shared_ptr<Node>
82 clone_with_new_inputs(
const OutputVector& new_args)
const override;
92 void set_is_foldable(
bool is_foldable) { m_is_foldable = is_foldable; }
93 bool get_is_foldable()
const {
return m_is_foldable; }
94 bool evaluate(
const HostTensorVector& output_values,
95 const HostTensorVector& input_values)
const override;
96 bool evaluate_lower(
const HostTensorVector& output_values)
const override;
97 bool evaluate_upper(
const HostTensorVector& output_values)
const override;
98 bool constant_fold(OutputVector& output_values,
99 const OutputVector& input_values)
override;
102 bool m_is_foldable =
true;
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
Definition: element_type.hpp:61
Root of all actual ops.
Definition: op.hpp:29
Operation that returns the shape of its input argument as a tensor.
Definition: shape_of.hpp:73
ShapeOf(const Output< Node > &arg)
Constructs a shape-of operation.
bool evaluate(const HostTensorVector &output_values, const HostTensorVector &input_values) const override
Evaluates the op on input_values putting results in output_values.
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
Operation that returns the shape of its input argument as a tensor.
Definition: shape_of.hpp:29
const NodeTypeInfo & get_type_info() const override
Definition: shape_of.hpp:32
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
ShapeOf(const Output< Node > &arg, const element::Type output_type=element::i64)
Constructs a shape-of operation.
bool evaluate(const HostTensorVector &output_values, const HostTensorVector &input_values) const override
Evaluates the op on input_values putting results in output_values.
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28