Home
last modified time | relevance | path

Searched refs:GetComputationCallers (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcall_graph.h252 std::vector<HloInstruction*> GetComputationCallers(HloComputation* c);
Dar_crs_combiner.cc196 auto caller_instructions = call_graph_->GetComputationCallers(computation); in WhileFromBodyParameter()
210 auto caller_instructions = call_graph_->GetComputationCallers(computation); in ConditionalFromBodyParameter()
Dcall_graph.cc369 std::vector<HloInstruction*> CallGraph::GetComputationCallers( in GetComputationCallers() function in xla::CallGraph
Dwhile_loop_all_reduce_code_motion.cc540 call_graph->GetComputationCallers(computation); in Run()