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};
57 virtual std::shared_ptr<Node>
58 clone_with_new_inputs(
const OutputVector& new_args)
const override;
A handle for one of a node's outputs.
Definition: node_output.hpp:42
Returns embeddings for given indices.
Definition: embeddingbag_packed_base.hpp:30
Returns embeddings for given indices.
Definition: embeddingbag_packedsum.hpp:31
EmbeddingBagPackedSum()=default
Constructs a EmbeddingBagPackedSum operation.
EmbeddingBagPackedSum(const Output< Node > &emb_table, const Output< Node > &indices, const Output< Node > &per_sample_weights)
Constructs a EmbeddingBagPackedSum operation.
const NodeTypeInfo & get_type_info() const override
Definition: embeddingbag_packedsum.hpp:34
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28