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};
70 virtual std::shared_ptr<Node>
71 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_offsets_base.hpp:30
Returns embeddings for given indices.
Definition: embeddingbag_offsets_sum.hpp:31
EmbeddingBagOffsetsSum()=default
Constructs a EmbeddingBagOffsetsSum operation.
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.
const NodeTypeInfo & get_type_info() const override
Definition: embeddingbag_offsets_sum.hpp:34
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28