Searched refs:prod_dev (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | buf_rendezvous.cc | 58 return strings::StrCat("[dev:", (prod_dev ? prod_dev->name() : "none"), in DebugString()
90 h->prod_dev = dev; in ProvideBuf()
|
| D | buf_rendezvous.h | 60 Device* prod_dev; member
67 : prod_dev(nullptr), in Hook()
|
| D | buf_rendezvous_test.cc | 66 EXPECT_EQ(h->prod_dev, fake_dev_ptr_); in TEST_F()
89 EXPECT_EQ(h->prod_dev, fake_dev_ptr_); in TEST_F()
|
| D | collective_rma_local.cc | 52 hook->prod_dev, // src Device 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 | 584 hook->prod_dev->attributes().device_type() == "CPU" || in RecvBufAsync()
597 hook->prod_value, "empty_name", hook->prod_dev, cpu_tensor, in RecvBufAsync()
|