Searched refs:call_graph_ (Results 1 – 14 of 14) sorted by relevance
63 call_graph_->NearestAncestorsInSameComputation( in GetExecutionConstraint()83 if (call_graph_->InstructionIsNestedIn(a, condition) && in GetExecutionConstraint()84 call_graph_->InstructionIsNestedIn(b, body)) { in GetExecutionConstraint()99 if (call_graph_->InstructionIsNestedIn( in GetExecutionConstraint()103 if (call_graph_->InstructionIsNestedIn( in GetExecutionConstraint()177 call_graph_->InstructionIsNestedIn(b.defining_instruction(), in IsDefinedBefore()182 call_graph_->InstructionIsNestedIn(a.defining_instruction(), in IsDefinedBefore()190 (call_graph_->InstructionIsNestedIn( in IsDefinedBefore()192 call_graph_->InstructionIsNestedIn( in IsDefinedBefore()202 if (call_graph_->InstructionIsNestedIn( in IsDefinedBefore()[all …]
41 : module_(module), call_graph_(CallGraph::Build(module)) {} in HloOrdering()102 const CallGraph& call_graph() const { return *call_graph_; } in call_graph()119 std::unique_ptr<CallGraph> call_graph_; variable
272 : module_(module), call_graph_(CallGraph::Build(&module)) {} in HloLivenessAnalysis()313 call_graph_.get()); in RunAnalysis()326 &worklist, &workset, call_graph_.get()); in RunAnalysis()
61 std::unique_ptr<CallGraph> call_graph_; variable
156 const CallGraph& call_graph() const { return *call_graph_; } in call_graph()318 std::unique_ptr<CallGraph> call_graph_; variable
195 auto caller_instructions = call_graph_->GetComputationCallers(computation); in WhileFromBodyParameter()209 auto caller_instructions = call_graph_->GetComputationCallers(computation); in ConditionalFromBodyParameter()342 combiner.call_graph_ = CallGraph::Build(module); in TestInstructionsComputeSameValue()606 call_graph_ = CallGraph::Build(module); in Run()
190 std::unique_ptr<CallGraph> call_graph_; variable
195 std::unique_ptr<CallGraph> call_graph_; variable
94 call_graph_(CallGraph::Build(&module)), in HloDataflowAnalysis()891 call_graph_->GetNode(parameter->parent()); in UpdateParameterValueSet()1279 call_graph_->GetNode(called_computation); in Propagate()1294 call_graph_->GetNode(instruction->parent()); in Propagate()1324 const CallGraphNode& call_graph_node = call_graph_->GetNode(computation); in InitializeInstructionValueSets()
1871 call_graph_->GetNode(instruction->parent()).GetCallSite(instruction); in CalledComputationsMemoryUsage()1933 const CallGraphNode& call_graph_node = call_graph_->GetNode(computation); in RematerializeComputation()2122 call_graph_ = CallGraph::Build(module); in Run()2123 TF_RETURN_IF_ERROR(call_graph_->VisitNodes( in Run()
623 std::unique_ptr<CallGraph> call_graph_; variable
219 call_graph_(std::move(call_graph)) {} in MemorySpaceAssignmentCostAnalysis()226 std::unique_ptr<CallGraph> call_graph_; variable
1822 CallGraphNode& node = call_graph_->GetNode(buffer.instruction()->parent()); in PropagateBufferConstraintToUses()2417 call_graph_ = CallGraph::Build(module); in Run()2435 CallGraphNode& node = call_graph_->GetNode(computation); in Run()2704 call_graph_ = CallGraph::Build(module); in Init()
360 auto& node = call_graph_->GetNode(computation); in CalculateComputationNestLevel()