19 #include "ngraph/axis_set.hpp"
20 #include "ngraph/op/util/embeddingbag_packed_base.hpp"
21 #include "ngraph/op/util/index_reduction.hpp"
33 static constexpr NodeTypeInfo type_info{
"EmbeddingBagPackedSum", 3};
34 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
52 const Output<Node>& indices,
53 const Output<Node>& per_sample_weights);
57 virtual std::shared_ptr<Node>
58 clone_with_new_inputs(
const OutputVector& new_args)
const override;
const NodeTypeInfo & get_type_info() const override
Definition: embeddingbag_packedsum.hpp:34
EmbeddingBagPackedSum(const Output< Node > &emb_table, const Output< Node > &indices, const Output< Node > &per_sample_weights)
Constructs a EmbeddingBagPackedSum operation.
Returns embeddings for given indices.
Definition: embeddingbag_packed_base.hpp:30
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28
Returns embeddings for given indices.
Definition: embeddingbag_packedsum.hpp:31
EmbeddingBagPackedSum()=default
Constructs a EmbeddingBagPackedSum operation.