Searched refs:control_predecessors (Results 1 – 18 of 18) sorted by relevance
100 for (HloInstruction* predecessor : hlo->control_predecessors()) { in Build()152 inputs.insert(inputs.end(), item->control_predecessors().begin(), in UpdateReachabilityThroughInstruction()153 item->control_predecessors().end()); in UpdateReachabilityThroughInstruction()
72 ASSERT_EQ(1, computation->root_instruction()->control_predecessors().size()); in TEST_F()73 auto prior = computation->root_instruction()->control_predecessors()[0]; in TEST_F()
36 changed = changed || !instruction->control_predecessors().empty(); in Run()
57 for (HloInstruction* control_predecessor : hlo->control_predecessors()) { in DefaultAction()
176 EXPECT_THAT(rng1->control_predecessors(), ElementsAre(rng0)); in TEST_F()
206 !instruction->control_predecessors().empty() || in TryHoistingInvariantInstructionsFromWhileBody()
199 if (!instruction->control_predecessors().empty() || in IsRemovable()260 TF_RET_CHECK(instruction->control_predecessors().empty()) in RemoveInstruction()376 for (HloInstruction* op : inst->control_predecessors()) { in ComputeInstructionPostOrder()
283 for (const HloInstruction* pred : instruction->control_predecessors()) { in Verify()
95 for (HloInstruction* control_predecessor : hlo->control_predecessors()) { in GlobalPredecessors()
428 while (!instruction->control_predecessors().empty()) { in StripControlDependenciesFrom()430 instruction->control_predecessors().front()->RemoveControlDependencyTo( in StripControlDependenciesFrom()
175 EXPECT_THAT(computation->root_instruction()->control_predecessors(), in TEST_F()
293 instruction->control_predecessors().empty()) { in CreateSchedule()
494 auto predecessors = cloned_add->control_predecessors(); in TEST_F()
1058 for (auto predecessor : best->control_predecessors()) { in RematerializeComputation()
889 const std::vector<HloInstruction*>& control_predecessors() const { in control_predecessors() function
1219 for (const HloInstruction* pred : instr->control_predecessors()) { in AddInstructionIncomingEdges()
1664 for (auto* ctrl_pred : inst->control_predecessors()) { in CopyAllControlDepsFrom()2763 for (HloInstruction* child : current_node->control_predecessors()) { in PostOrderDFS()
95 hlo->control_predecessors().begin(), in GpuHloOrdering()96 hlo->control_predecessors().end()); in GpuHloOrdering()