Added copy_if for Paul

5d612a82 · wsttiger · d0c3a01d · 5d612a82
Commit 5d612a82 authored Sep 26, 2018 by wsttiger
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 5 deletions

src/include/migraph/operators.hpp src/include/migraph/operators.hpp +4 -5

No files found.
--- a/src/include/migraph/operators.hpp
+++ b/src/include/migraph/operators.hpp
@@ -375,11 +375,10 @@ struct squeeze
        std::vector<std::size_t> new_lens;
        if(axes.empty())
        {
-            for (unsigned long len : old_lens)
+            std::copy_if(old_lens.begin(),
-            {
+                         old_lens.end(),
-                if(len != 1)
+                         std::back_inserter(new_lens),
-                    new_lens.push_back(len);
+                         [](auto len) { return len != 1; });
-            }
        }
        else
        {