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()284 const CallGraphNode& call_graph_node = in ComputeConditionalAliasedBuffers() local287 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in ComputeConditionalAliasedBuffers()290 CHECK_EQ(call_graph_node.caller_callsites().size(), 1); in ComputeConditionalAliasedBuffers()
779 const CallGraphNode& call_graph_node = in UpdateParameterValueSet() local784 if (call_graph_node.context() == CallContext::kParallel || in UpdateParameterValueSet()785 call_graph_node.caller_callsites().empty()) { in UpdateParameterValueSet()788 CHECK_EQ(call_graph_node.context(), CallContext::kSequential); in UpdateParameterValueSet()792 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in UpdateParameterValueSet()1188 const CallGraphNode& call_graph_node = in Propagate() local1190 if (call_graph_node.context() == CallContext::kSequential) { in Propagate()1203 const CallGraphNode& call_graph_node = in Propagate() local1205 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in Propagate()1235 const CallGraphNode& call_graph_node = call_graph_->GetNode(computation); in InitializeInstructionValueSets() local[all …]
1881 const CallGraphNode& call_graph_node = call_graph_->GetNode(computation); in RematerializeComputation() local1959 const CallSite* callsite = call_graph_node.GetCallSite(instruction); in RematerializeComputation()