Searched refs:prod_ctx (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/core/common_runtime/ |
D | buf_rendezvous.h | 61 DeviceContext* prod_ctx; member 68 prod_ctx(nullptr), in Hook()
|
D | buf_rendezvous.cc | 59 ", ctx:", reinterpret_cast<uint64>(prod_ctx), in DebugString() 91 h->prod_ctx = dev_ctx; in ProvideBuf()
|
D | buf_rendezvous_test.cc | 67 EXPECT_EQ(h->prod_ctx, fake_dev_ctx_); in TEST_F() 90 EXPECT_EQ(h->prod_ctx, fake_dev_ctx_); in TEST_F()
|
D | collective_rma_local.cc | 50 MemCpyAsync(hook->prod_ctx, // src DeviceContext in RecvFromPeer()
|
/external/tensorflow/tensorflow/contrib/gdr/ |
D | gdr_worker.cc | 177 hook->prod_dev, hook->prod_ctx, hook->prod_attr.on_host(), in RecvBufAsync()
|
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
D | grpc_worker_service.cc | 596 hook->prod_ctx->CopyDeviceTensorToCPU( in RecvBufAsync()
|