19 #include "ngraph/axis_set.hpp"
20 #include "ngraph/op/util/index_reduction.hpp"
32 static constexpr
NodeTypeInfo type_info{
"EmbeddingSegmentsSum", 3};
78 virtual std::shared_ptr<Node>
79 clone_with_new_inputs(
const OutputVector& new_args)
const override;
81 virtual bool visit_attributes(
AttributeVisitor& visitor)
override {
return true; }
83 static constexpr
int EMB_TABLE = 0;
84 static constexpr
int INDICES = 1;
85 static constexpr
int SEGMENT_IDS = 2;
86 static constexpr
int NUM_SEGMENTS = 3;
87 static constexpr
int DEFAULT_INDEX = 4;
88 static constexpr
int PER_SAMPLE_WEIGHTS = 5;
91 using v3::EmbeddingSegmentsSum;
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
Returns embeddings for given indices.
Definition: embedding_segments_sum.hpp:30
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
EmbeddingSegmentsSum()=default
Constructs a EmbeddingSegmentsSum operation.
EmbeddingSegmentsSum(const Output< Node > &emb_table, const Output< Node > &indices, const Output< Node > &segment_ids, const Output< Node > &num_segments, const Output< Node > &default_index, const Output< Node > &per_sample_weights)
Constructs a EmbeddingSegmentsSum operation.
const NodeTypeInfo & get_type_info() const override
Definition: embedding_segments_sum.hpp:33
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28