Searched refs:prod_value (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | buf_rendezvous.h | 65 const Tensor* prod_value; member
72 prod_value(nullptr), in Hook()
|
| D | collective_rma_local.cc | 69 CHECK_EQ(recv_bytes, hook->prod_value->TotalBytes()); in RecvFromPeer()
76 hook->prod_value, // src Tensor* in RecvFromPeer()
|
| D | buf_rendezvous.cc | 67 ", val:", reinterpret_cast<uint64>(prod_value), in DebugString()
99 h->prod_value = v; in ProvideBuf()
|
| D | buf_rendezvous_test.cc | 101 EXPECT_EQ(h->prod_value, &a_); in TEST_F()
125 EXPECT_EQ(h->prod_value, &a_); in TEST_F()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 646 if (!DMAHelper::CanUseDMA(hook->prod_value)) { in RecvBufAsync()
661 const size_t num_bytes = hook->prod_value->TotalBytes(); in RecvBufAsync()
674 hook->prod_value->dtype(), hook->prod_value->shape()); in RecvBufAsync()
676 hook->prod_value, "empty_name", hook->prod_dev, cpu_tensor, in RecvBufAsync()
690 rendezvous_done(*hook->prod_value, s); in RecvBufAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed_test.cc | 112 int64 num_bytes = h->prod_value->TotalBytes(); in RecvBufAsync()
114 reinterpret_cast<const char*>(DMAHelper::base(h->prod_value)), in RecvBufAsync()
|