Searched refs:rhs_tree (Results 1 – 1 of 1) sorted by relevance
207 const ShapeTree<HloSharding>& rhs_tree) { in AssignTreeSharding() argument209 auto rhs_it = rhs_tree.begin(); in AssignTreeSharding()210 for (; lhs_it != lhs_tree->end() && rhs_it != rhs_tree.end(); in AssignTreeSharding()213 if (rhs_tree.IsLeaf(rhs_it->first)) { in AssignTreeSharding()227 TF_RET_CHECK(rhs_it == rhs_tree.end()); in AssignTreeSharding()