Home
last modified time | relevance | path

Searched refs:inverse_permutation (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dreshape_mover.cc155 std::vector<int64> inverse_permutation = in UpdateOperand() local
158 new_shape, operand, inverse_permutation)); in UpdateOperand()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral.cc871 std::vector<int64> inverse_permutation = InversePermutation(permutation); in Transpose() local
876 layout->add_minor_to_major(inverse_permutation[index]); in Transpose()
880 new_literal.SetDynamicSize(inverse_permutation[i], GetDynamicSize(i)); in Transpose()