Searched refs:prod_value (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | buf_rendezvous.h | 62 const Tensor* prod_value; member
69 prod_value(nullptr), in Hook()
|
| D | collective_rma_local.cc | 49 CHECK_EQ(recv_bytes, hook->prod_value->TotalBytes()); in RecvFromPeer()
56 hook->prod_value, // src Tensor* in RecvFromPeer()
|
| D | buf_rendezvous.cc | 60 ", val:", reinterpret_cast<uint64>(prod_value), in DebugString()
92 h->prod_value = v; in ProvideBuf()
|
| D | buf_rendezvous_test.cc | 68 EXPECT_EQ(h->prod_value, &a_); in TEST_F()
91 EXPECT_EQ(h->prod_value, &a_); in TEST_F()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 573 if (!DMAHelper::CanUseDMA(hook->prod_value)) { in RecvBufAsync()
582 const size_t num_bytes = hook->prod_value->TotalBytes(); in RecvBufAsync()
594 hook->prod_value->dtype(), in RecvBufAsync()
595 hook->prod_value->shape()); in RecvBufAsync()
597 hook->prod_value, "empty_name", hook->prod_dev, cpu_tensor, in RecvBufAsync()
613 SetTensorInRecvBufResp(recv_buf_max_chunk_, hook->prod_value, in RecvBufAsync()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_worker.cc | 168 if (!DMAHelper::CanUseDMA(hook->prod_value)) { in RecvBufAsync()
176 response->mutable_transport_options(), *hook->prod_value, in RecvBufAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed_test.cc | 107 int64 num_bytes = h->prod_value->TotalBytes(); in RecvBufAsync()
109 reinterpret_cast<const char*>(DMAHelper::base(h->prod_value)), in RecvBufAsync()
|