Searched refs:HasReplicatedSharding (Results 1 – 3 of 3) sorted by relevance
40 bool HasReplicatedSharding(const HloSharding& sharding);
47 bool HasReplicatedSharding(const HloSharding& sharding) { in HasReplicatedSharding() function49 return absl::c_any_of(sharding.tuple_elements(), HasReplicatedSharding); in HasReplicatedSharding()
3739 TF_RET_CHECK(!HasReplicatedSharding(hlo->sharding()) || in PreprocessSharding()