Home
last modified time | relevance | path

Searched refs:IsPermutation (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dutil.cc111 bool IsPermutation(absl::Span<const int64> permutation, int64 rank) { in IsPermutation() function
122 DCHECK(IsPermutation(input_permutation, input_permutation.size())); in InversePermutation()
Dutil.h331 bool IsPermutation(absl::Span<const int64> permutation, int64 rank);
344 CHECK(IsPermutation(permutation, data.size())); in Permute()
Dliteral.cc672 CHECK(IsPermutation(permutation, shape().rank())) in Transpose()
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc2955 if (!IsPermutation(dimensions, operand.rank())) { in InferTransposeShape()
Dalgebraic_simplifier.cc1736 CHECK(IsPermutation(permutation, permutation.size())); in OptimizeDotOfReorderContractingDims()