Aligh default reduction with base

mmikolajcz · May 17, 2024 · f457c07 · f457c07
1 parent 2d57b4d
commit f457c07
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/core/include/openvino/op/embeddingbag_offsets.hpp b/src/core/include/openvino/op/embeddingbag_offsets.hpp
@@ -41,18 +41,18 @@ class OPENVINO_API EmbeddingBagOffsets : public util::EmbeddingBagOffsetsBase {
                         const Output<Node>& offsets,
                         const Output<Node>& default_index,
                         const Output<Node>& per_sample_weights,
-                        const Reduction& reduction = Reduction::MEAN);
+                        const Reduction& reduction = Reduction::SUM);
 
     EmbeddingBagOffsets(const Output<Node>& emb_table,
                         const Output<Node>& indices,
                         const Output<Node>& offsets,
                         const Output<Node>& default_index,
-                        const Reduction& reduction = Reduction::MEAN);
+                        const Reduction& reduction = Reduction::SUM);
 
     EmbeddingBagOffsets(const Output<Node>& emb_table,
                         const Output<Node>& indices,
                         const Output<Node>& offsets,
-                        const Reduction& reduction = Reduction::MEAN);
+                        const Reduction& reduction = Reduction::SUM);
 
     std::shared_ptr<Node> clone_with_new_inputs(const OutputVector& new_args) const override;
 };

diff --git a/src/core/include/openvino/op/embeddingbag_packed.hpp b/src/core/include/openvino/op/embeddingbag_packed.hpp
@@ -35,11 +35,11 @@ class OPENVINO_API EmbeddingBagPacked : public util::EmbeddingBagPackedBase {
     EmbeddingBagPacked(const Output<Node>& emb_table,
                        const Output<Node>& indices,
                        const Output<Node>& per_sample_weights,
-                       const Reduction& reduction = Reduction::MEAN);
+                       const Reduction& reduction = Reduction::SUM);
 
     EmbeddingBagPacked(const Output<Node>& emb_table,
                        const Output<Node>& indices,
-                       const Reduction& reduction = Reduction::MEAN);
+                       const Reduction& reduction = Reduction::SUM);
 
     std::shared_ptr<Node> clone_with_new_inputs(const OutputVector& new_args) const override;
 };