Searched refs:prod_ctx (Results 1 – 5 of 5) sorted by relevance
65 DeviceContext* prod_ctx; member75 prod_ctx(nullptr), in Hook()
80 ", ctx:", reinterpret_cast<uint64>(prod_ctx), in DebugString()117 h->prod_ctx = dev_ctx; in ProvideBuf()
105 EXPECT_EQ(h->prod_ctx, fake_device_context_); in TEST_F()131 EXPECT_EQ(h->prod_ctx, fake_device_context_); in TEST_F()371 EXPECT_EQ(h->prod_ctx, fake_device_context_); in TEST_F()
67 MemCpyAsync(hook->prod_ctx, // src DeviceContext in RecvFromPeer()
678 hook->prod_ctx->CopyDeviceTensorToCPU( in RecvBufAsync()