Update ngraph/core/reference/src/runtime/reference/concat.cpp

Co-authored-by: Gleb Kazantaev <[email protected]>
openvinotoolkit · Feb 11, 2021 · 70fae5c · 70fae5c
1 parent a33d2f1
commit 70fae5c
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/ngraph/core/reference/src/runtime/reference/concat.cpp b/ngraph/core/reference/src/runtime/reference/concat.cpp
@@ -30,10 +30,10 @@ namespace ngraph
                 {
                     std::vector<size_t> sizes;
                     sizes.reserve(in_shapes.size());
-                    for (size_t in_index = 0; in_index != in_shapes.size(); ++in_index)
-                    {
-                        sizes.push_back(shape_size(in_shapes[in_index]));
-                    }
+                    std::transform(begin(in_shapes),
+                                   end(in_shapes),
+                                   std::back_inserter(sizes),
+                                   [](const Shape& shape) { return shape_size(shape); });
                     return sizes;
                 }
             } // namespace
@@ -69,6 +69,6 @@ namespace ngraph
                     }
                 }
             }
-        }
-    }
-}
+        } // namespace reference
+    }     // namespace runtime
+} // namespace ngraph