Searched refs:a_ancestor (Results 1 – 2 of 2) sorted by relevance
60 const HloInstruction* a_ancestor; in GetExecutionConstraint() local62 std::tie(a_ancestor, b_ancestor) = in GetExecutionConstraint()66 if (a_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()81 const HloComputation* body = a_ancestor->while_body(); in GetExecutionConstraint()82 const HloComputation* condition = a_ancestor->while_condition(); in GetExecutionConstraint()94 if (a_ancestor == b_ancestor && in GetExecutionConstraint()95 (a_ancestor->opcode() == HloOpcode::kConditional)) { in GetExecutionConstraint()98 for (int j = 0; j < a_ancestor->branch_count(); ++j) { in GetExecutionConstraint()[all …]
401 HloInstruction* a_ancestor = a; in NearestAncestorsInSameComputation() local411 a_ancestor = next_caller(a_ancestor); in NearestAncestorsInSameComputation()412 if (a_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()430 a_ancestor = next_caller(a_ancestor); in NearestAncestorsInSameComputation()