Searched refs:padded_src_shape (Results 1 – 1 of 1) sorted by relevance
415 auto padded_src_shape = in TileToPartialReplicateHaloExchange() local432 padded_src_shape.dimensions(dim) / dst_shard_count; in TileToPartialReplicateHaloExchange()490 padded_src_shape.dimensions(dim) / in TileToPartialReplicateHaloExchange()508 auto padded_src_shape = in PadFromPartialReplicateShape() local528 padded_src_shape.dimensions(dim) / src_shard_count; in PadFromPartialReplicateShape()584 padded_src_shape.dimensions(dim) - max_right_halo_size)); in PadFromPartialReplicateShape()617 padded_dst_shape.dimensions(dim) - padded_src_shape.dimensions(dim)); in PadFromPartialReplicateShape()620 padded_src_shape.dimensions(dim)); in PadFromPartialReplicateShape()