Searched refs:PadToShape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.h | 116 HloInstruction* PadToShape(HloInstruction* hlo, const Shape& padded_shape,
|
| D | spmd_partitioner_util.cc | 250 HloInstruction* PadToShape(HloInstruction* hlo, const Shape& padded_shape, in PadToShape() function
300 return PadToShape(hlo, padded_base_shape, b); in PadBaseShapeBeforeUnevenTiledSharding()
|
| D | dot_handler.cc | 3235 auto padded_operand = PadToShape(outside_to_inside[broadcast->operand(0)], in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3253 PadToShape(outside_to_inside[constant], in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
|
| D | spmd_partitioner.cc | 1147 auto padded_hlo = PadToShape(hlo_, padded_shape, state_.b); in ReshardWithAllToAll()
2604 return PadToShape(infeed_element, pad_shape, &branch_b); in HandleInfeed()
|