Home
last modified time | relevance | path

Searched refs:GetCallSite (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcall_graph_test.cc233 EXPECT_EQ(entry_node.GetCallSite(call), &call_callsite); in TEST_F()
240 EXPECT_EQ(entry_node.GetCallSite(map), &map_callsite); in TEST_F()
283 EXPECT_EQ(entry_node.GetCallSite(conditional), &conditional_callsite); in TEST_F()
Dcall_graph.h108 const CallSite* GetCallSite(const HloInstruction* instruction) const;
Dcall_graph.cc88 const CallSite* CallGraphNode::GetCallSite( in GetCallSite() function in xla::CallGraphNode
Dhlo_rematerialization.cc959 call_graph_->GetNode(instruction->parent()).GetCallSite(instruction); in CalledComputationsMemoryUsage()
1131 const CallSite* callsite = call_graph_node.GetCallSite(instruction); in RematerializeComputation()