Searched refs:tuple_elements_ (Results 1 – 2 of 2) sorted by relevance
90 parts.reserve(tuple_elements_.size()); in ToString()91 for (const HloSharding& element : tuple_elements_) { in ToString()110 return absl::c_any_of(tuple_elements_, [&](const HloSharding& s) { in UsesDevice()210 TF_RET_CHECK(shape_leaves == tuple_elements_.size()) in CheckLeafCount()212 << " leaf nodes while this sharding has " << tuple_elements_.size(); in CheckLeafCount()221 auto it = tuple_elements_.begin(); in AsShapeTree()246 if (tuple_elements_.empty()) { in UniqueDevice()250 for (auto& tuple_sharding : tuple_elements_) { in UniqueDevice()411 for (const HloSharding& element : tuple_elements_) { in ToProto()459 auto begin_it = tuple_elements_.begin() + sharding_index; in GetSubSharding()[all …]
105 tuple_elements_, [](const HloSharding& s) { return s.IsReplicated(); }); in IsReplicated()113 return absl::c_all_of(tuple_elements_, [](const HloSharding& s) { in IsTileMaximal()191 tuple_elements_ == other.tuple_elements_;211 return tuple_elements_; in tuple_elements()245 tuple_elements_(tuple_shardings) {}280 std::vector<HloSharding> tuple_elements_; variable