Searched refs:map_callsite (Results 1 – 1 of 1) sorted by relevance
235 const CallSite& map_callsite = entry_node.callsites()[1]; in TEST_F() local236 EXPECT_EQ(map, map_callsite.instruction()); in TEST_F()237 EXPECT_THAT(map_callsite.called_computations(), in TEST_F()239 EXPECT_EQ(CallContext::kParallel, map_callsite.context()); in TEST_F()240 EXPECT_EQ(entry_node.GetCallSite(map), &map_callsite); in TEST_F()