Searched refs:ApplyDomainShardingPass (Results 1 – 1 of 1) sorted by relevance
315 StatusOr<int64> ApplyDomainShardingPass(const DomainMetadata::Domain& domain, in ApplyDomainShardingPass() function352 TF_RETURN_IF_ERROR(ApplyDomainShardingPass(domain, sharding).status()); in ApplyDomainSharding()