Commit 12c86c6d authored by Khalique's avatar Khalique
Browse files

changed swap for matrix transpose

parent b6c92026
...@@ -356,7 +356,7 @@ struct tf_parser ...@@ -356,7 +356,7 @@ struct tf_parser
std::vector<int64_t> perm(args[0]->get_shape().lens().size()); std::vector<int64_t> perm(args[0]->get_shape().lens().size());
std::iota(perm.begin(), perm.end(), int64_t{0}); std::iota(perm.begin(), perm.end(), int64_t{0});
// swap the last two elements // swap the last two elements
std::swap(*perm.rbegin(), *(perm.rbegin() + 1)); std::iter_swap(perm.end() - 1, perm.end() - 2);
auto l1 = (transa) ? prog.add_instruction(op::transpose{perm}, args[0]) : args[0]; auto l1 = (transa) ? prog.add_instruction(op::transpose{perm}, args[0]) : args[0];
auto l2 = (transb) ? prog.add_instruction(op::transpose{perm}, args[1]) : args[1]; auto l2 = (transb) ? prog.add_instruction(op::transpose{perm}, args[1]) : args[1];
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment