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()
610 const CallGraphNode& call_graph_node = in UpdateParameterValueSet() local615 if (call_graph_node.context() == CallContext::kParallel || in UpdateParameterValueSet()616 call_graph_node.caller_callsites().empty()) { in UpdateParameterValueSet()619 CHECK_EQ(call_graph_node.context(), CallContext::kSequential); in UpdateParameterValueSet()623 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in UpdateParameterValueSet()889 const CallGraphNode& call_graph_node = in Propagate() local891 if (call_graph_node.context() == CallContext::kSequential) { in Propagate()904 const CallGraphNode& call_graph_node = in Propagate() local906 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in Propagate()936 const CallGraphNode& call_graph_node = call_graph_->GetNode(computation); in InitializeInstructionValueSets() local[all …]
1869 const CallGraphNode& call_graph_node = call_graph_->GetNode(computation); in RematerializeComputation() local1932 const CallSite* callsite = call_graph_node.GetCallSite(instruction); in RematerializeComputation()