Home
last modified time | relevance | path

Searched refs:PermuteDimensions (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h486 static Shape PermuteDimensions(tensorflow::gtl::ArraySlice<int64> permutation,
Dshape_util.cc971 /* static */ Shape ShapeUtil::PermuteDimensions( in PermuteDimensions() function in xla::ShapeUtil
Dliteral_util.cc775 ShapeUtil::PermuteDimensions(inverse_permutation, shape()); in Transpose()
/external/tensorflow/tensorflow/compiler/xla/service/
Duser_computation.cc3191 ShapeUtil::PermuteDimensions( in Visit()
3206 ShapeUtil::PermuteDimensions( in Visit()
Dalgebraic_simplifier.cc963 ShapeUtil::PermuteDimensions({1, 0}, dot->shape()), in HandleDot()
Dshape_inference.cc2342 return ShapeUtil::PermuteDimensions(InversePermutation(dimensions), operand); in InferTransposeShape()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc676 ShapeUtil::PermuteDimensions( in AreShapesForTranspose021()