Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dgather_expander.cc49 return MakeTransposeHlo(start_indices, permutation); in TransposeIndexVectorDimToLast()
269 return MakeTransposeHlo(accumulator, permutation); in PermuteBatchAndOffsetDims()
Dscatter_expander.cc51 return MakeTransposeHlo(scatter_indices, permutation); in TransposeIndexVectorDimToLast()
110 return MakeTransposeHlo(updates, permutation); in PermuteScatterAndWindowDims()
Dhlo_creation_utils.h64 StatusOr<HloInstruction*> MakeTransposeHlo(HloInstruction* operand,
Dhlo_creation_utils.cc97 StatusOr<HloInstruction*> MakeTransposeHlo(HloInstruction* operand, in MakeTransposeHlo() function
Dalgebraic_simplifier.cc310 return MakeTransposeHlo(dot_operand, transpose_dimensions); in NormalizeDotOperandToBatchMajorAndContractingMinor()