Home
last modified time | relevance | path

Searched refs:InversePermutation (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization.cc139 InversePermutation(new_output_dim_order)))); in Run()
/external/tensorflow/tensorflow/compiler/xla/service/
Dreshape_mover.cc209 InversePermutation(first_reshape_operand->dimensions()); in UpdateOperand()
Dlayout_assignment.cc1026 auto inverse_dimensions = InversePermutation(user->dimensions()); in ChooseOutputLayoutFromOperandLayout()
Duser_computation.cc3192 InversePermutation(AsInt64Slice(reshape_request.dimensions())), in Visit()
3207 InversePermutation(AsInt64Slice(transpose_request.dimensions())), in Visit()
Dshape_inference.cc2342 return ShapeUtil::PermuteDimensions(InversePermutation(dimensions), operand); in InferTransposeShape()
/external/tensorflow/tensorflow/compiler/xla/
Dutil.cc175 std::vector<int64> InversePermutation( in InversePermutation() function
Dutil.h280 std::vector<int64> InversePermutation(
Dliteral_util.cc773 std::vector<int64> inverse_permutation = InversePermutation(permutation); in Transpose()