Merge branch 'develop' into dynamic_reduce

de6a9ba6 · Brian Pickrell · GitHub · 0fe131d2 · 48cc33e4 · de6a9ba6
Unverified Commit de6a9ba6 authored Dec 06, 2022 by Brian Pickrell Committed by GitHub Dec 06, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 8 deletions

src/include/migraphx/op/squeeze.hpp src/include/migraphx/op/squeeze.hpp +4 -8

No files found.
--- a/src/include/migraphx/op/squeeze.hpp
+++ b/src/include/migraphx/op/squeeze.hpp
@@ -70,14 +70,10 @@ struct squeeze
            std::vector<shape::dynamic_dimension> dyn_dims = {};
            if(axes.empty())
            {
-                for(auto i : range(input_shape.ndim()))
+                std::copy_if(input_shape.dyn_dims().cbegin(),
-                {
+                             input_shape.dyn_dims().cend(),
-                    auto dd = input_shape.dyn_dims()[i];
+                             std::back_inserter(dyn_dims),
-                    if(dd != one_dyn_dim)
+                             [&](auto dd) { return dd != one_dyn_dim; });
-                    {
-                        dyn_dims.push_back(dd);
-                    }
-                }
            }
            else
            {