diff options
-rw-r--r-- | tensorflow/compiler/xla/literal_util.cc | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/tensorflow/compiler/xla/literal_util.cc b/tensorflow/compiler/xla/literal_util.cc index f48f0eb7a5..7091c324d1 100644 --- a/tensorflow/compiler/xla/literal_util.cc +++ b/tensorflow/compiler/xla/literal_util.cc @@ -279,8 +279,7 @@ void TransposeLiteralInternal(const Literal& original, for (int64 i = 0; i < indices.size(); ++i) { new_indices[i] = indices[permutation[i]]; } - LiteralUtil::Set<T>( - result, Permute(InversePermutation(permutation), indices), value); + LiteralUtil::Set<T>(result, new_indices, value); }); } } // namespace |