Searched refs:GetCallSite (Results 1 – 4 of 4) sorted by relevance
233 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()
108 const CallSite* GetCallSite(const HloInstruction* instruction) const;
88 const CallSite* CallGraphNode::GetCallSite( in GetCallSite() function in xla::CallGraphNode
959 call_graph_->GetNode(instruction->parent()).GetCallSite(instruction); in CalledComputationsMemoryUsage()1131 const CallSite* callsite = call_graph_node.GetCallSite(instruction); in RematerializeComputation()