Home
last modified time | relevance | path

Searched refs:call_graph_ (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_ordering.cc44 call_graph_->NearestAncestorsInSameComputation( in ExecutesBefore()
63 if (call_graph_->InstructionIsNestedIn(a, condition) && in ExecutesBefore()
64 call_graph_->InstructionIsNestedIn(b, body)) { in ExecutesBefore()
79 if (call_graph_->InstructionIsNestedIn( in ExecutesBefore()
83 if (call_graph_->InstructionIsNestedIn( in ExecutesBefore()
134 call_graph_->InstructionIsNestedIn(b.defining_instruction(), in IsDefinedBefore()
139 call_graph_->InstructionIsNestedIn(a.defining_instruction(), in IsDefinedBefore()
147 (call_graph_->InstructionIsNestedIn( in IsDefinedBefore()
149 call_graph_->InstructionIsNestedIn( in IsDefinedBefore()
159 if (call_graph_->InstructionIsNestedIn( in IsDefinedBefore()
[all …]
Dhlo_ordering.h41 : module_(module), call_graph_(CallGraph::Build(module)) {} in HloOrdering()
72 const CallGraph& call_graph() const { return *call_graph_; } in call_graph()
89 std::unique_ptr<CallGraph> call_graph_; variable
Dhlo_dataflow_analysis.h134 const CallGraph& call_graph() const { return *call_graph_; } in call_graph()
225 std::unique_ptr<CallGraph> call_graph_; variable
Dhlo_liveness_analysis.cc253 : module_(module), call_graph_(CallGraph::Build(&module)) {} in HloLivenessAnalysis()
294 call_graph_.get()); in RunAnalysis()
307 &worklist, &workset, call_graph_.get()); in RunAnalysis()
Dar_crs_combiner.cc99 auto caller_instructions = call_graph_->GetComputationCallers(computation); in WhileFromBodyParameter()
173 combiner.call_graph_ = CallGraph::Build(module); in TestInstructionsComputeSameValue()
378 call_graph_ = CallGraph::Build(module); in Run()
Dhlo_liveness_analysis.h60 std::unique_ptr<CallGraph> call_graph_; variable
Dhlo_rematerialization.h113 std::unique_ptr<CallGraph> call_graph_; variable
Dar_crs_combiner.h149 std::unique_ptr<CallGraph> call_graph_; variable
Dhlo_dataflow_analysis.cc50 call_graph_(CallGraph::Build(&module)), in HloDataflowAnalysis()
472 call_graph_->GetNode(parameter->parent()); in UpdateParameterValueSet()
691 call_graph_->GetNode(called_computation); in Propagate()
706 call_graph_->GetNode(instruction->parent()); in Propagate()
737 const CallGraphNode& call_graph_node = call_graph_->GetNode(computation); in InitializeInstructionValueSets()
Dhlo_rematerialization.cc959 call_graph_->GetNode(instruction->parent()).GetCallSite(instruction); in CalledComputationsMemoryUsage()
1005 const CallGraphNode& call_graph_node = call_graph_->GetNode(computation); in RematerializeComputation()
1237 call_graph_ = CallGraph::Build(module); in Run()
1238 TF_RETURN_IF_ERROR(call_graph_->VisitNodes( in Run()