Searched refs:parallel_sharding (Results 1 – 1 of 1) sorted by relevance
971 absl::optional<HloSharding> parallel_sharding; in GatherDataOperandShardingFromOutput() local977 parallel_sharding = in GatherDataOperandShardingFromOutput()991 return parallel_sharding; in GatherDataOperandShardingFromOutput()993 if (!parallel_sharding) { in GatherDataOperandShardingFromOutput()996 if (MergeSharding(*parallel_sharding, &*passthrough_sharding, in GatherDataOperandShardingFromOutput()1000 if (MergeSharding(*passthrough_sharding, &*parallel_sharding, in GatherDataOperandShardingFromOutput()1002 return parallel_sharding; in GatherDataOperandShardingFromOutput()