19 #include "ngraph/op/op.hpp"
44 const bool with_right_bound =
true);
49 virtual std::shared_ptr<Node>
50 clone_with_new_inputs(
const OutputVector& inputs)
const override;
52 element::Type get_output_type()
const {
return m_output_type; }
53 void set_output_type(
element::Type output_type) { m_output_type = output_type; }
55 using Node::set_output_type;
57 bool get_with_right_bound()
const {
return m_with_right_bound; }
58 void set_with_right_bound(
bool with_right_bound)
60 m_with_right_bound = with_right_bound;
64 element::Type m_output_type;
65 bool m_with_right_bound;
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 bucketizes the input based on boundaries.
Definition: bucketize.hpp:29
virtual void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
Bucketize(const Output< Node > &data, const Output< Node > &buckets, const element::Type output_type=element::i64, const bool with_right_bound=true)
Constructs a Bucketize node.
const NodeTypeInfo & get_type_info() const override
Definition: bucketize.hpp:32
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28