Searched refs:domain_sharding (Results 1 – 1 of 1) sorted by relevance
233 const HloSharding& domain_sharding) { in ApplyShardingFromUsers() argument239 domain_sharding.tuple_elements().size()); in ApplyShardingFromUsers()240 instruction->set_sharding(domain_sharding); in ApplyShardingFromUsers()260 instruction->set_sharding(domain_sharding); in ApplyShardingFromUsers()316 const HloSharding& domain_sharding) { in ApplyDomainShardingPass() argument330 ApplyShardingFromUsers(instruction, domain, domain_sharding)); in ApplyDomainShardingPass()