Searched refs:b_ancestor (Results 1 – 2 of 2) sorted by relevance
402 HloInstruction* b_ancestor = b; in NearestAncestorsInSameComputation() local418 b_ancestor = next_caller(b_ancestor); in NearestAncestorsInSameComputation()419 if (b_ancestor == nullptr) { in NearestAncestorsInSameComputation()425 while ((a_ancestor != nullptr) && (b_ancestor != nullptr)) { in NearestAncestorsInSameComputation()426 if (a_ancestor->parent() == b_ancestor->parent()) { in NearestAncestorsInSameComputation()427 return {a_ancestor, b_ancestor}; in NearestAncestorsInSameComputation()431 b_ancestor = next_caller(b_ancestor); in NearestAncestorsInSameComputation()
61 const HloInstruction* b_ancestor; in GetExecutionConstraint() local62 std::tie(a_ancestor, b_ancestor) = in GetExecutionConstraint()73 CHECK_NE(b_ancestor, nullptr); in GetExecutionConstraint()74 CHECK_EQ(a_ancestor->parent(), b_ancestor->parent()); in GetExecutionConstraint()80 if (a_ancestor == b_ancestor && a_ancestor->opcode() == HloOpcode::kWhile) { in GetExecutionConstraint()94 if (a_ancestor == b_ancestor && in GetExecutionConstraint()111 CHECK_EQ(b, b_ancestor); in GetExecutionConstraint()133 if (ExecutesBeforeInSameComputation(a_ancestor, b_ancestor)) { in GetExecutionConstraint()136 if (ExecutesBeforeInSameComputation(b_ancestor, a_ancestor)) { in GetExecutionConstraint()