Home
last modified time | relevance | path

Searched defs:padded_hlo (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dfft_handler.cc383 auto padded_hlo = PadEachPartitionWithHaloExchange( in HandleFft() local
Dspmd_partitioner.cc468 auto padded_hlo = in ReshardNoCache() local
755 auto padded_hlo = ShapeUtil::Compatible(padded_shape, base_shape_) in ReshardAsWindowedInput() local
1016 auto padded_hlo = PadFromPartialReplicateShape( in ReshardFromPartialReplicateWithDynamicSlice() local
1147 auto padded_hlo = PadToShape(hlo_, padded_shape, state_.b); in ReshardWithAllToAll() local
Dspmd_partitioner_util.cc1724 auto padded_hlo = PadBaseShapeBeforeUnevenTiledSharding( in PerGroupSliceFromReplicated() local