19 #include "ngraph/axis_set.hpp"
20 #include "ngraph/op/util/embeddingbag_offsets_base.hpp"
21 #include "ngraph/op/util/index_reduction.hpp"
33 static constexpr NodeTypeInfo type_info{
"EmbeddingBagOffsetsSum", 3};
34 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
56 const Output<Node>& indices,
57 const Output<Node>& offsets,
58 const Output<Node>& default_index,
59 const Output<Node>& per_sample_weights);
62 const Output<Node>& indices,
63 const Output<Node>& offsets,
64 const Output<Node>& default_index);
67 const Output<Node>& indices,
68 const Output<Node>& offsets);
70 virtual std::shared_ptr<Node>
71 clone_with_new_inputs(
const OutputVector& new_args)
const override;
EmbeddingBagOffsetsSum(const Output< Node > &emb_table, const Output< Node > &indices, const Output< Node > &offsets, const Output< Node > &default_index, const Output< Node > &per_sample_weights)
Constructs a EmbeddingBagOffsetsSum operation.
EmbeddingBagOffsetsSum()=default
Constructs a EmbeddingBagOffsetsSum operation.
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28
Returns embeddings for given indices.
Definition: embeddingbag_offsets_sum.hpp:31
Returns embeddings for given indices.
Definition: embeddingbag_offsets_base.hpp:30
const NodeTypeInfo & get_type_info() const override
Definition: embeddingbag_offsets_sum.hpp:34