Searched refs:prod_dev (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | buf_rendezvous.cc | 65 return absl::StrCat("[dev:", (prod_dev ? prod_dev->name() : "none"), in DebugString()
97 h->prod_dev = dev; in ProvideBuf()
|
| D | buf_rendezvous.h | 63 Device* prod_dev; member
70 : prod_dev(nullptr), in Hook()
|
| D | buf_rendezvous_test.cc | 99 EXPECT_EQ(h->prod_dev, default_device_); in TEST_F()
123 EXPECT_EQ(h->prod_dev, default_device_); in TEST_F()
|
| D | collective_rma_local.cc | 72 hook->prod_dev, // src Device in RecvFromPeer()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 663 hook->prod_dev->attributes().device_type() == "CPU" || in RecvBufAsync()
676 hook->prod_value, "empty_name", hook->prod_dev, cpu_tensor, in RecvBufAsync()
|