Searched refs:MakeTransposeHlo (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | gather_expander.cc | 49 return MakeTransposeHlo(start_indices, permutation); in TransposeIndexVectorDimToLast()
269 return MakeTransposeHlo(accumulator, permutation); in PermuteBatchAndOffsetDims()
|
| D | scatter_expander.cc | 51 return MakeTransposeHlo(scatter_indices, permutation); in TransposeIndexVectorDimToLast()
110 return MakeTransposeHlo(updates, permutation); in PermuteScatterAndWindowDims()
|
| D | hlo_creation_utils.h | 73 StatusOr<HloInstruction*> MakeTransposeHlo(HloInstruction* operand,
|
| D | space_to_batch_converter.cc | 519 MakeTransposeHlo(activations, transpose_dims)); in BringSpaceNextToBatch()
553 MakeTransposeHlo(activations, transpose_dims)); in BringSpaceNextToBatch()
1752 MakeTransposeHlo(output_slice, transpose_dims)); in BatchToSpace()
2124 MakeTransposeHlo(consumer, reversed_transpose_dims)); in PropagateOnConstant()
|
| D | hlo_creation_utils.cc | 111 StatusOr<HloInstruction*> MakeTransposeHlo(HloInstruction* operand, in MakeTransposeHlo() function
|
| D | algebraic_simplifier.cc | 413 return MakeTransposeHlo(dot_operand, transpose_dimensions); in NormalizeDotOperandToBatchMajorAndContractingMinor()
|