Home
last modified time | relevance | path

Searched refs:a_ancestor (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_ordering.cc41 const HloInstruction* a_ancestor; in ExecutesBefore() local
43 std::tie(a_ancestor, b_ancestor) = in ExecutesBefore()
47 if (a_ancestor == nullptr) { in ExecutesBefore()
54 CHECK_EQ(a_ancestor->parent(), b_ancestor->parent()); in ExecutesBefore()
60 if (a_ancestor == b_ancestor && a_ancestor->opcode() == HloOpcode::kWhile) { in ExecutesBefore()
61 const HloComputation* body = a_ancestor->while_body(); in ExecutesBefore()
62 const HloComputation* condition = a_ancestor->while_condition(); in ExecutesBefore()
74 if (a_ancestor == b_ancestor && in ExecutesBefore()
75 (a_ancestor->opcode() == HloOpcode::kConditional)) { in ExecutesBefore()
78 for (int j = 0; j < a_ancestor->branch_count(); ++j) { in ExecutesBefore()
[all …]
Dcall_graph.cc393 HloInstruction* a_ancestor = a; in NearestAncestorsInSameComputation() local
403 a_ancestor = next_caller(a_ancestor); in NearestAncestorsInSameComputation()
404 if (a_ancestor == nullptr) { in NearestAncestorsInSameComputation()
417 while ((a_ancestor != nullptr) && (b_ancestor != nullptr)) { in NearestAncestorsInSameComputation()
418 if (a_ancestor->parent() == b_ancestor->parent()) { in NearestAncestorsInSameComputation()
419 return {a_ancestor, b_ancestor}; in NearestAncestorsInSameComputation()
422 a_ancestor = next_caller(a_ancestor); in NearestAncestorsInSameComputation()
Dbuffer_assignment.cc758 const HloInstruction* a_ancestor; in MayInterfereAcrossSubcomputations() local
760 std::tie(a_ancestor, b_ancestor) = in MayInterfereAcrossSubcomputations()
763 if (a_ancestor == nullptr) { in MayInterfereAcrossSubcomputations()
767 if (a_ancestor->opcode() == HloOpcode::kWhile && in MayInterfereAcrossSubcomputations()
769 a_ancestor->while_body())) { in MayInterfereAcrossSubcomputations()
772 a_ancestor->operand(0)); in MayInterfereAcrossSubcomputations()
779 assignment->liveness().points_to_analysis().GetPointsToSet(a_ancestor); in MayInterfereAcrossSubcomputations()