Searched refs:per_input_sharding (Results 1 – 1 of 1) sorted by relevance
2703 auto per_input_sharding = hlo->sharding(); in HandleReduce() local2707 per_input_sharding = hlo->sharding().GetSubSharding(hlo->shape(), {0}); in HandleReduce()