Searched refs:call_graph_node (Results 1 – 5 of 5) sorted by relevance
99 [&](const CallGraphNode& call_graph_node) -> Status { in Run() argument100 if (call_graph_node.computation()->IsFusionComputation()) { in Run()101 TF_RET_CHECK(call_graph_node.caller_callsites().size() == 1); in Run()103 call_graph_node.caller_callsites()[0].instruction(); in Run()
199 const CallGraphNode& call_graph_node = in PropagateLivenessToParameterCallers() local201 if (call_graph_node.context() == CallContext::kSequential) { in PropagateLivenessToParameterCallers()202 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in PropagateLivenessToParameterCallers()229 const CallGraphNode& call_graph_node = in PropagateLivenessThroughControlFlow() local231 if (call_graph_node.context() == CallContext::kSequential) { in PropagateLivenessThroughControlFlow()232 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in PropagateLivenessThroughControlFlow()
229 const CallGraphNode& call_graph_node = in ComputeWhileAliasedBuffers() local231 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in ComputeWhileAliasedBuffers()234 CHECK_EQ(call_graph_node.caller_callsites().size(), 1); in ComputeWhileAliasedBuffers()248 const CallGraphNode& call_graph_node = in ComputeWhileAliasedBuffers() local251 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in ComputeWhileAliasedBuffers()255 CHECK_EQ(call_graph_node.caller_callsites().size(), 1); in ComputeWhileAliasedBuffers()283 const CallGraphNode& call_graph_node = in ComputeConditionalAliasedBuffers() local286 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in ComputeConditionalAliasedBuffers()289 CHECK_EQ(call_graph_node.caller_callsites().size(), 1); in ComputeConditionalAliasedBuffers()
471 const CallGraphNode& call_graph_node = in UpdateParameterValueSet() local476 if (call_graph_node.context() == CallContext::kParallel || in UpdateParameterValueSet()477 call_graph_node.caller_callsites().empty()) { in UpdateParameterValueSet()480 CHECK_EQ(call_graph_node.context(), CallContext::kSequential); in UpdateParameterValueSet()484 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in UpdateParameterValueSet()690 const CallGraphNode& call_graph_node = in Propagate() local692 if (call_graph_node.context() == CallContext::kSequential) { in Propagate()705 const CallGraphNode& call_graph_node = in Propagate() local707 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in Propagate()737 const CallGraphNode& call_graph_node = call_graph_->GetNode(computation); in InitializeInstructionValueSets() local[all …]
1005 const CallGraphNode& call_graph_node = call_graph_->GetNode(computation); in RematerializeComputation() local1131 const CallSite* callsite = call_graph_node.GetCallSite(instruction); in RematerializeComputation()