Searched refs:GatherParallelDataOperandSharding (Results 1 – 1 of 1) sorted by relevance
1011 std::optional<HloSharding> GatherParallelDataOperandSharding( in GatherParallelDataOperandSharding() function1093 GatherParallelDataOperandSharding(output_sharding, hlo, *parallel_dims); in GatherDataOperandShardingFromOutput()