Searched refs:MakeTransposeHlo (Results 1 – 5 of 5) sorted by relevance
49 return MakeTransposeHlo(start_indices, permutation); in TransposeIndexVectorDimToLast()269 return MakeTransposeHlo(accumulator, permutation); in PermuteBatchAndOffsetDims()
51 return MakeTransposeHlo(scatter_indices, permutation); in TransposeIndexVectorDimToLast()110 return MakeTransposeHlo(updates, permutation); in PermuteScatterAndWindowDims()
64 StatusOr<HloInstruction*> MakeTransposeHlo(HloInstruction* operand,
97 StatusOr<HloInstruction*> MakeTransposeHlo(HloInstruction* operand, in MakeTransposeHlo() function
310 return MakeTransposeHlo(dot_operand, transpose_dimensions); in NormalizeDotOperandToBatchMajorAndContractingMinor()