Searched refs:prod_value (Results 1 – 6 of 6) sorted by relevance
| /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 "dynamic", hook->prod_value->dtype(), &hook->prod_value->shape()); in RecvBufAsync()
677 hook->prod_value->dtype(), hook->prod_value->shape()); in RecvBufAsync()
679 hook->prod_value, "empty_name", hook->prod_dev, cpu_tensor, in RecvBufAsync()
693 rendezvous_done(*hook->prod_value, s); in RecvBufAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | buf_rendezvous.h | 66 const Tensor* prod_value; member
76 prod_value(nullptr), in Hook()
|
| D | collective_rma_local.cc | 66 CHECK_EQ(recv_bytes, hook->prod_value->TotalBytes()); in RecvFromPeer()
73 hook->prod_value, // src Tensor* in RecvFromPeer()
|
| D | buf_rendezvous.cc | 81 ", val:", reinterpret_cast<uint64>(prod_value), in DebugString()
118 h->prod_value = v; in ProvideBuf()
|
| D | buf_rendezvous_test.cc | 106 EXPECT_EQ(h->prod_value, &a_); in TEST_F()
132 EXPECT_EQ(h->prod_value, &a_); in TEST_F()
372 EXPECT_EQ(h->prod_value, &a_); in TEST_F()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_rma_distributed_test.cc | 121 int64 num_bytes = h->prod_value->TotalBytes(); in RecvBufAsync()
123 reinterpret_cast<const char*>(DMAHelper::base(h->prod_value)), in RecvBufAsync()
|