Searched refs:caller_callsites (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | flatten_call_graph_test.cc | 147 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()
|
| D | defuser.cc | 101 TF_RET_CHECK(call_graph_node.caller_callsites().size() == 1); in Run()
103 call_graph_node.caller_callsites()[0].instruction(); in Run()
|
| D | flatten_call_graph.cc | 74 for (int i = 0; i < node.caller_callsites().size(); ++i) { in FlattenNode()
75 CallSite call_site = node.caller_callsites()[i]; in FlattenNode()
|
| D | hlo_alias_analysis.cc | 231 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()
287 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in ComputeConditionalAliasedBuffers()
290 CHECK_EQ(call_graph_node.caller_callsites().size(), 1); in ComputeConditionalAliasedBuffers()
|
| D | call_graph.cc | 362 node.caller_callsites().size() > 1) { in IsFlattened()
372 for (const auto& callsite : GetNode(c).caller_callsites()) { in GetComputationCallers()
388 if (node.caller_callsites().size() != 1) { in NearestAncestorsInSameComputation()
391 return node.caller_callsites()[0].instruction(); in NearestAncestorsInSameComputation()
|
| D | call_graph_test.cc | 109 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()
|
| D | hlo_liveness_analysis.cc | 202 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in PropagateLivenessToParameterCallers()
232 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in PropagateLivenessThroughControlFlow()
|
| D | call_inliner.cc | 149 .caller_callsites() in Run()
|
| D | call_graph.h | 114 const std::vector<CallSite>& caller_callsites() const { in caller_callsites() function
|
| D | hlo_dataflow_analysis.cc | 616 call_graph_node.caller_callsites().empty()) { in UpdateParameterValueSet()
623 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in UpdateParameterValueSet()
906 for (const CallSite& callsite : call_graph_node.caller_callsites()) { in Propagate()
988 if (call_graph_node.caller_callsites().empty() || in InitializeInstructionValueSets()
|
| D | layout_assignment.cc | 1583 if (node.caller_callsites().size() != 1) { in PropagateBufferConstraintToUses()
1586 const HloInstruction* parent = node.caller_callsites()[0].instruction(); in PropagateBufferConstraintToUses()
2120 if (node.caller_callsites().size() == 1) { in Run()
2123 if (absl::c_none_of(node.caller_callsites(), [](CallSite caller) { in Run()
2128 for (int64 i = 0; i < node.caller_callsites().size() - 1; ++i) { in Run()
2129 HloInstruction* caller = node.caller_callsites()[i].instruction(); in Run()
|
| D | memory_space_assignment.cc | 237 auto callsites = node.caller_callsites(); in CalculateComputationNestLevel()
|