Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Ddot_decomposer.cc208 ShapeUtil::PermuteDimensions(InversePermutation(lhs_transpose), in CanonicalizeDot()
252 ShapeUtil::PermuteDimensions(InversePermutation(rhs_transpose), in CanonicalizeDot()
Dreshape_mover.cc155 InversePermutation(first_reshape_operand->dimensions()); in UpdateOperand()
Dlayout_assignment.cc1100 auto inverse_dimensions = InversePermutation(user->dimensions()); in ChooseOutputLayoutFromOperandLayout()
Dshape_inference.cc2732 return ShapeUtil::PermuteDimensions(InversePermutation(dimensions), operand); in InferTransposeShape()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dconv_canonicalization.cc142 InversePermutation(new_output_dim_order)))); in Run()
/external/tensorflow/tensorflow/compiler/xla/
Dutil.cc89 std::vector<int64> InversePermutation( in InversePermutation() function
Dutil.h325 std::vector<int64> InversePermutation(
Dshape_util_test.cc708 InversePermutation(permutation))); in TEST()
Dshape_util.cc980 CHECK(TransposeIsBitcast(shape, new_shape, InversePermutation(permutation))) in PermuteDimensions()
Dliteral.cc684 std::vector<int64> inverse_permutation = InversePermutation(permutation); in Transpose()