Searched defs:output_shard_shape (Results 1 – 1 of 1) sorted by relevance
2704 auto output_shard_shape = MakePartitionedShape(hlo->shape(), sharding); in HandleReshape() local2959 auto output_shard_shape = MakePartitionedShape(hlo->shape(), hlo->sharding()); in HandleBroadcast() local