Commit 68727a64 authored by Paul's avatar Paul
Browse files

Merge branch 'dot-add' into bert-opt2

parents a81f593b 56615a84
......@@ -78,7 +78,7 @@ shape transpose_batch(const shape& s, unsigned trans_batch)
std::vector<int64_t> perm(s.lens().size());
std::iota(perm.begin(), perm.end(), 0);
std::swap(perm[batch], perm[batch + trans_batch]);
return reorder_shape(s, perm);
return shape::from_permutation(s.type(), s.lens(), perm);
}
template <class R, class... Ts, class... Us>
......
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