Searched refs:single_sharding (Results 1 – 1 of 1) sorted by relevance
144 auto single_sharding = sharding->ExtractSingleSharding(); in CloneShardingForDomain() local145 if (!single_sharding) { in CloneShardingForDomain()148 return std::make_shared<const HloSharding>(*single_sharding); in CloneShardingForDomain()344 auto single_sharding = sharding.ExtractSingleSharding(); in ApplyDomainSharding() local345 if (single_sharding) { in ApplyDomainSharding()349 return ApplyDomainSingleSharding(domain, *single_sharding); in ApplyDomainSharding()