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