Searched refs:GatherParallelDataOperandSharding (Results 1 – 1 of 1) sorted by relevance
894 absl::optional<HloSharding> GatherParallelDataOperandSharding( in GatherParallelDataOperandSharding() function978 GatherParallelDataOperandSharding(hlo.sharding(), hlo, *parallel_dims); in GatherDataOperandShardingFromOutput()