New Gather op reference implementation.

openvinotoolkit · Dec 16, 2020 · 9947bbe · 9947bbe
1 parent 61752b8
commit 9947bbe
Show file tree

Hide file tree

Showing 3 changed files with 585 additions and 165 deletions.
diff --git a/ngraph/core/reference/include/ngraph/coordinate_range.hpp b/ngraph/core/reference/include/ngraph/coordinate_range.hpp
@@ -154,6 +154,7 @@ namespace ngraph
                 bool increment();
 
                 bool is_valid() const noexcept { return !has_zeros(m_source_shape); }
+
             private:
                 const Shape m_source_shape;
                 const CoordinateBounds m_bounds;
@@ -185,7 +186,7 @@ namespace ngraph
                              Strides(source_shape.size(), 1));
             }
 
-            /// \brief Class allows to iterate over Tensor with reverted axies part by part.
+            /// \brief Class allows to iterate over Tensor with reverted axes part by part.
             ///
             /// To create ReverseRange use _reverse_ function.
             ///
@@ -200,6 +201,7 @@ namespace ngraph
                 bool increment();
 
                 bool is_valid() const noexcept { return !has_zeros(m_source_shape); }
+
             private:
                 const Shape m_source_shape;
                 const std::vector<size_t> m_memory_strides;
@@ -213,8 +215,14 @@ namespace ngraph
                 return ReverseRange(source_shape, reversed_axis);
             }
 
+            inline ReverseRange index(const Shape& source_shape)
+            {
+                return reverse(source_shape, {});
+            }
+
         } // namespace impl
         using impl::Direction;
+        using impl::index;
         using impl::reverse;
         using impl::slice;
     } // namespace coordinates