19 #include "ngraph/op/op.hpp"
31 static constexpr NodeTypeInfo type_info{
"Bucketize", 3};
32 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
42 const Output<Node>& buckets,
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;
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.
Definition: element_type.hpp:61
const NodeTypeInfo & get_type_info() const override
Definition: bucketize.hpp:32
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:70
Operation that bucketizes the input based on boundaries.
Definition: bucketize.hpp:29
Root of all actual ops.
Definition: op.hpp:29