Searched refs:call_graph_node (Results 1 – 5 of 5) sorted by relevance
100 [&](const CallGraphNode& call_graph_node) -> Status { in Run() argument101 if (call_graph_node.computation()->IsFusionComputation()) { in Run()102 TF_RET_CHECK(call_graph_node.caller_callsites().size() == 1); in Run()104 call_graph_node.caller_callsites()[0].instruction(); in Run()
92 const CallGraphNode& call_graph_node = in ComputeWhileAliasedValues() local94 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in ComputeWhileAliasedValues()97 CHECK_EQ(call_graph_node.caller_callsites().size(), 1); in ComputeWhileAliasedValues()113 const CallGraphNode& call_graph_node = in ComputeWhileAliasedValues() local116 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in ComputeWhileAliasedValues()120 CHECK_EQ(call_graph_node.caller_callsites().size(), 1) in ComputeWhileAliasedValues()146 const CallGraphNode& call_graph_node = in ComputeConditionalAliasedValues() local148 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in ComputeConditionalAliasedValues()151 CHECK_EQ(call_graph_node.caller_callsites().size(), 1); in ComputeConditionalAliasedValues()
195 const CallGraphNode& call_graph_node = in PropagateLivenessToParameterCallers() local197 if (call_graph_node.context() == CallContext::kControlFlow) { in PropagateLivenessToParameterCallers()198 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in PropagateLivenessToParameterCallers()224 const CallGraphNode& call_graph_node = in PropagateLivenessThroughControlFlow() local226 if (call_graph_node.context() == CallContext::kControlFlow) { in PropagateLivenessThroughControlFlow()227 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in PropagateLivenessThroughControlFlow()
890 const CallGraphNode& call_graph_node = in UpdateParameterValueSet() local895 if (call_graph_node.context() == CallContext::kEmbedded || in UpdateParameterValueSet()896 call_graph_node.caller_callsites().empty()) { in UpdateParameterValueSet()899 CHECK_EQ(call_graph_node.context(), CallContext::kControlFlow); in UpdateParameterValueSet()903 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in UpdateParameterValueSet()1278 const CallGraphNode& call_graph_node = in Propagate() local1280 if (call_graph_node.context() == CallContext::kControlFlow) { in Propagate()1293 const CallGraphNode& call_graph_node = in Propagate() local1295 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in Propagate()1304 } else if (call_graph_node.context() == CallContext::kControlFlow) { in Propagate()[all …]
1933 const CallGraphNode& call_graph_node = call_graph_->GetNode(computation); in RematerializeComputation() local2012 const CallSite* callsite = call_graph_node.GetCallSite(instruction); in RematerializeComputation()