Home
last modified time | relevance | path

Searched refs:caller_callsites (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dflatten_call_graph_test.cc147 EXPECT_EQ(1, c_node.caller_callsites().size()); in TEST_F()
182 EXPECT_EQ(2, cond_node.caller_callsites().size()); in TEST_F()
190 EXPECT_EQ(1, cond_node.caller_callsites().size()); in TEST_F()
221 EXPECT_EQ(1, c_node.caller_callsites().size()); in TEST_F()
224 EXPECT_EQ(1, b_node.caller_callsites().size()); in TEST_F()
254 EXPECT_EQ(1, sub_node.caller_callsites().size()); in TEST_F()
Ddefuser.cc101 TF_RET_CHECK(call_graph_node.caller_callsites().size() == 1); in Run()
103 call_graph_node.caller_callsites()[0].instruction(); in Run()
Dhlo_alias_analysis.cc231 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in ComputeWhileAliasedBuffers()
234 CHECK_EQ(call_graph_node.caller_callsites().size(), 1); in ComputeWhileAliasedBuffers()
251 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in ComputeWhileAliasedBuffers()
255 CHECK_EQ(call_graph_node.caller_callsites().size(), 1); in ComputeWhileAliasedBuffers()
286 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in ComputeConditionalAliasedBuffers()
289 CHECK_EQ(call_graph_node.caller_callsites().size(), 1); in ComputeConditionalAliasedBuffers()
Dflatten_call_graph.cc74 for (int i = 0; i < node.caller_callsites().size(); ++i) { in FlattenNode()
75 CallSite call_site = node.caller_callsites()[i]; in FlattenNode()
Dcall_graph.cc359 node.caller_callsites().size() > 1) { in IsFlattened()
369 for (auto callsite : GetNode(c).caller_callsites()) { in GetComputationCallers()
385 if (node.caller_callsites().size() != 1) { in NearestAncestorsInSameComputation()
388 return node.caller_callsites()[0].instruction(); in NearestAncestorsInSameComputation()
Dcall_graph_test.cc109 EXPECT_TRUE(node.caller_callsites().empty()); in TEST_F()
156 EXPECT_TRUE(entry_node.caller_callsites().empty()); in TEST_F()
165 EXPECT_EQ(5, map_node.caller_callsites().size()); in TEST_F()
190 EXPECT_TRUE(entry_node.caller_callsites().empty()); in TEST_F()
198 EXPECT_EQ(3, called_node.caller_callsites().size()); in TEST_F()
Dhlo_liveness_analysis.cc202 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in PropagateLivenessToParameterCallers()
232 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in PropagateLivenessThroughControlFlow()
Dcall_inliner.cc145 for (const CallSite& callsite : node.caller_callsites()) { in Run()
Dhlo_dataflow_analysis.cc477 call_graph_node.caller_callsites().empty()) { in UpdateParameterValueSet()
484 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in UpdateParameterValueSet()
707 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in Propagate()
796 if (call_graph_node.caller_callsites().empty() || in InitializeInstructionValueSets()
Dcall_graph.h114 const std::vector<CallSite>& caller_callsites() const { in caller_callsites() function
Dcopy_insertion.cc1222 !node.caller_callsites().empty()) { in IsWhileBody()
1225 CHECK_EQ(node.caller_callsites().size(), 1); in IsWhileBody()
1227 node.caller_callsites()[0].instruction(); in IsWhileBody()