Searched defs:padded_hlo (Results 1 – 3 of 3) sorted by relevance
383 auto padded_hlo = PadEachPartitionWithHaloExchange( in HandleFft() local
468 auto padded_hlo = in ReshardNoCache() local755 auto padded_hlo = ShapeUtil::Compatible(padded_shape, base_shape_) in ReshardAsWindowedInput() local1016 auto padded_hlo = PadFromPartialReplicateShape( in ReshardFromPartialReplicateWithDynamicSlice() local1147 auto padded_hlo = PadToShape(hlo_, padded_shape, state_.b); in ReshardWithAllToAll() local
1724 auto padded_hlo = PadBaseShapeBeforeUnevenTiledSharding( in PerGroupSliceFromReplicated() local