change the reshape implementation.

17a6cdcf · Shucai Xiao · 9124bec7 · 17a6cdcf
Commit 17a6cdcf authored Jan 22, 2019 by Shucai Xiao
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 5 deletions

src/include/migraphx/operators.hpp src/include/migraphx/operators.hpp +5 -5

No files found.
--- a/src/include/migraphx/operators.hpp
+++ b/src/include/migraphx/operators.hpp
@@ -598,11 +598,11 @@ struct reshape
                    rdims[i] = missing_dim;
            }
        }
-        if(dims.back() == -1)
+        //if(dims.back() == -1)
-        {
+        //{
-            rdims.pop_back();
+        //    rdims.pop_back();
-            std::copy(idims.begin() + rdims.size(), idims.end(), std::back_inserter(rdims));
+        //    std::copy(idims.begin() + rdims.size(), idims.end(), std::back_inserter(rdims));
-        }
+        //}
        shape s{inputs.front().type(), rdims};
        if(s.elements() != inputs.front().elements())
            MIGRAPHX_THROW("Wrong number of elements for reshape");