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