Home
last modified time | relevance | path

Searched refs:control_predecessors (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_reachability.cc100 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()
Dcall_inliner_test.cc72 ASSERT_EQ(1, computation->root_instruction()->control_predecessors().size()); in TEST_F()
73 auto prior = computation->root_instruction()->control_predecessors()[0]; in TEST_F()
Ddespecializer.cc36 changed = changed || !instruction->control_predecessors().empty(); in Run()
Dcall_inliner.cc57 for (HloInstruction* control_predecessor : hlo->control_predecessors()) { in DefaultAction()
Dhlo_element_type_converter_test.cc176 EXPECT_THAT(rng1->control_predecessors(), ElementsAre(rng0)); in TEST_F()
Dwhile_loop_invariant_code_motion.cc206 !instruction->control_predecessors().empty() || in TryHoistingInvariantInstructionsFromWhileBody()
Dhlo_computation.cc199 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()
Dhlo_schedule.cc283 for (const HloInstruction* pred : instruction->control_predecessors()) { in Verify()
Dhlo_module_group_util.cc95 for (HloInstruction* control_predecessor : hlo->control_predecessors()) { in GlobalPredecessors()
Dcopy_insertion.cc428 while (!instruction->control_predecessors().empty()) { in StripControlDependenciesFrom()
430 instruction->control_predecessors().front()->RemoveControlDependencyTo( in StripControlDependenciesFrom()
Dwhile_loop_simplifier_test.cc175 EXPECT_THAT(computation->root_instruction()->control_predecessors(), in TEST_F()
Dhlo_memory_scheduler.cc293 instruction->control_predecessors().empty()) { in CreateSchedule()
Dhlo_computation_test.cc494 auto predecessors = cloned_add->control_predecessors(); in TEST_F()
Dhlo_rematerialization.cc1058 for (auto predecessor : best->control_predecessors()) { in RematerializeComputation()
Dhlo_instruction.h889 const std::vector<HloInstruction*>& control_predecessors() const { in control_predecessors() function
Dhlo_graph_dumper.cc1219 for (const HloInstruction* pred : instr->control_predecessors()) { in AddInstructionIncomingEdges()
Dhlo_instruction.cc1664 for (auto* ctrl_pred : inst->control_predecessors()) { in CopyAllControlDepsFrom()
2763 for (HloInstruction* child : current_node->control_predecessors()) { in PostOrderDFS()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_hlo_schedule.cc95 hlo->control_predecessors().begin(), in GpuHloOrdering()
96 hlo->control_predecessors().end()); in GpuHloOrdering()