Searched refs:passthrough_sharding (Results 1 – 1 of 1) sorted by relevance
983 absl::optional<HloSharding> passthrough_sharding = in GatherDataOperandShardingFromOutput() local990 if (!passthrough_sharding) { in GatherDataOperandShardingFromOutput()994 return passthrough_sharding; in GatherDataOperandShardingFromOutput()996 if (MergeSharding(*parallel_sharding, &*passthrough_sharding, in GatherDataOperandShardingFromOutput()998 return passthrough_sharding; in GatherDataOperandShardingFromOutput()1000 if (MergeSharding(*passthrough_sharding, &*parallel_sharding, in GatherDataOperandShardingFromOutput()