Searched refs:control_successors (Results 1 – 13 of 13) sorted by relevance
898 std::any_of(candidate->control_successors().begin(), in PickRematerializationCandidate()899 candidate->control_successors().end(), in PickRematerializationCandidate()1055 for (auto successor : best->control_successors()) { in RematerializeComputation()1099 for (auto successor : remat->control_successors()) { in RematerializeComputation()
200 !instruction->control_successors().empty()) { in IsRemovable()263 TF_RET_CHECK(instruction->control_successors().empty()) in RemoveInstruction()788 instruction->control_successors().empty() && in CollectUnreachableRoots()1026 for (auto successor : instr->control_successors()) { in CloneWithReplacements()
160 for (const HloInstruction* succ : item->control_successors()) { in UpdateReachabilityThroughInstruction()
175 EXPECT_THAT(rng0->control_successors(), ElementsAre(rng1)); in TEST_F()
141 EXPECT_THAT(computation->root_instruction()->control_successors(), in TEST_F()
272 for (HloInstruction* succ : instruction->control_successors()) { in CreateSchedule()335 for (HloInstruction* succ : best->control_successors()) { in CreateSchedule()
207 !instruction->control_successors().empty()) { in TryHoistingInvariantInstructionsFromWhileBody()
173 for (HloInstruction* control_successor : hlo->control_successors()) { in GlobalSuccessors()
423 while (!instruction->control_successors().empty()) { in StripControlDependenciesFrom()425 instruction->control_successors().front())); in StripControlDependenciesFrom()
497 auto successors = predecessors[0]->control_successors(); in TEST_F()
892 const std::vector<HloInstruction*>& control_successors() const { in control_successors() function
63 count += instruction->control_successors().size(); in CountControlEdges()
1668 for (auto* ctrl_succ : inst->control_successors()) { in CopyAllControlDepsFrom()