Home
last modified time | relevance | path

Searched refs:PadBaseShapeBeforeUnevenTiledSharding (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h126 HloInstruction* PadBaseShapeBeforeUnevenTiledSharding(
Dspmd_partitioner_util.cc293 HloInstruction* PadBaseShapeBeforeUnevenTiledSharding( in PadBaseShapeBeforeUnevenTiledSharding() function
1724 auto padded_hlo = PadBaseShapeBeforeUnevenTiledSharding( in PerGroupSliceFromReplicated()
Ddot_handler.cc773 slice_operand = PadBaseShapeBeforeUnevenTiledSharding( in PartitionBaseCase()
Dspmd_partitioner.cc469 PadBaseShapeBeforeUnevenTiledSharding(hlo_, target, state_.b); in ReshardNoCache()