19 #include "ngraph/axis_set.hpp"
20 #include "ngraph/op/util/index_reduction.hpp"
32 static constexpr NodeTypeInfo type_info{
"EmbeddingBagPackedBase", 3};
33 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
51 const Output<Node>& indices,
52 const Output<Node>& per_sample_weights);
60 static constexpr
int EMB_TABLE = 0;
61 static constexpr
int INDICES = 1;
62 static constexpr
int PER_SAMPLE_WEIGHTS = 2;
EmbeddingBagPackedBase()=default
Constructs a EmbeddingBagPackedBase operation.
const NodeTypeInfo & get_type_info() const override
Definition: embeddingbag_packed_base.hpp:33
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
Returns embeddings for given indices.
Definition: embeddingbag_packed_base.hpp:30
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28
EmbeddingBagPackedBase(const Output< Node > &emb_table, const Output< Node > &indices, const Output< Node > &per_sample_weights)
Constructs a EmbeddingBagPackedBase operation.
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:70
Root of all actual ops.
Definition: op.hpp:29