Home
last modified time | relevance | path

Searched refs:prod_ctx (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dbuf_rendezvous.h61 DeviceContext* prod_ctx; member
68 prod_ctx(nullptr), in Hook()
Dbuf_rendezvous.cc59 ", ctx:", reinterpret_cast<uint64>(prod_ctx), in DebugString()
91 h->prod_ctx = dev_ctx; in ProvideBuf()
Dbuf_rendezvous_test.cc67 EXPECT_EQ(h->prod_ctx, fake_dev_ctx_); in TEST_F()
90 EXPECT_EQ(h->prod_ctx, fake_dev_ctx_); in TEST_F()
Dcollective_rma_local.cc50 MemCpyAsync(hook->prod_ctx, // src DeviceContext in RecvFromPeer()
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_worker.cc177 hook->prod_dev, hook->prod_ctx, hook->prod_attr.on_host(), in RecvBufAsync()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_worker_service.cc596 hook->prod_ctx->CopyDeviceTensorToCPU( in RecvBufAsync()