Searched refs:parallel_sharding (Results 1 – 1 of 1) sorted by relevance
1087 std::optional<HloSharding> parallel_sharding; in GatherDataOperandShardingFromOutput() local1092 parallel_sharding = in GatherDataOperandShardingFromOutput()1102 return parallel_sharding; in GatherDataOperandShardingFromOutput()1104 if (!parallel_sharding) { in GatherDataOperandShardingFromOutput()1107 if (MergeSharding(*parallel_sharding, &*passthrough_sharding, in GatherDataOperandShardingFromOutput()1111 if (MergeSharding(*passthrough_sharding, &*parallel_sharding, in GatherDataOperandShardingFromOutput()1113 return parallel_sharding; in GatherDataOperandShardingFromOutput()1115 return parallel_sharding; in GatherDataOperandShardingFromOutput()