Searched refs:prod_ctx (Results 1 – 5 of 5) sorted by relevance
64 DeviceContext* prod_ctx; member71 prod_ctx(nullptr), in Hook()
66 ", ctx:", reinterpret_cast<uint64>(prod_ctx), in DebugString()98 h->prod_ctx = dev_ctx; in ProvideBuf()
100 EXPECT_EQ(h->prod_ctx, fake_device_context_); in TEST_F()124 EXPECT_EQ(h->prod_ctx, fake_device_context_); in TEST_F()
70 MemCpyAsync(hook->prod_ctx, // src DeviceContext in RecvFromPeer()
675 hook->prod_ctx->CopyDeviceTensorToCPU( in RecvBufAsync()