Searched defs:padded_hlo (Results 1 – 3 of 3) sorted by relevance
381 auto padded_hlo = PadEachPartitionWithHaloExchange( in HandleFft() local
556 auto padded_hlo = PadBaseShapeBeforeUnevenTiledSharding( in ReshardNoCache() local987 auto padded_hlo = in ReshardAsWindowedInput() local1412 auto padded_hlo = PadFromPartialReplicateShape( in ReshardFromPartialReplicateWithDynamicSlice() local1543 auto padded_hlo = PadToShape(hlo_, padded_shape, state_.b); in ReshardWithAllToAll() local
1611 auto padded_hlo = PadBaseShapeBeforeUnevenTiledSharding( in PerGroupSliceFromReplicated() local