Searched refs:src_dev (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_worker.cc | 64 Device* src_dev = nullptr; in GrpcRecvTensorAsync() local
66 s = PrepareRecvTensor(parsed, &src_dev); in GrpcRecvTensorAsync()
81 [this, opts, response, done, src_dev, dma_ok]( in GrpcRecvTensorAsync()
93 (src_dev->tensorflow_gpu_device_info() == nullptr) || in GrpcRecvTensorAsync()
106 transport_options, val, src_dev, send_args.device_context, in GrpcRecvTensorAsync()
119 if (src_dev->tensorflow_gpu_device_info() && (!on_host)) { in GrpcRecvTensorAsync()
125 Allocator* alloc = src_dev->GetAllocator(alloc_attrs); in GrpcRecvTensorAsync()
128 << "send dev name: " << src_dev->name() in GrpcRecvTensorAsync()
129 << " gpu_info: " << src_dev->tensorflow_gpu_device_info(); in GrpcRecvTensorAsync()
139 GPUUtil::CopyGPUTensorToCPU(src_dev, send_dev_context, &val, copy, in GrpcRecvTensorAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_worker_service.cc | 379 Device* src_dev = nullptr; in GrpcRecvTensorAsync() local
381 s = PrepareRecvTensor(parsed, &src_dev); in GrpcRecvTensorAsync()
395 [opts, response, done, src_dev](const Status& status, in GrpcRecvTensorAsync()
410 if (src_dev->tensorflow_gpu_device_info() && (!on_host)) { in GrpcRecvTensorAsync()
416 Allocator* alloc = src_dev->GetAllocator(alloc_attrs); in GrpcRecvTensorAsync()
419 << "send dev name: " << src_dev->name() in GrpcRecvTensorAsync()
420 << " gpu_info: " << src_dev->tensorflow_gpu_device_info(); in GrpcRecvTensorAsync()
430 GPUUtil::CopyGPUTensorToCPU(src_dev, send_dev_context, &val, copy, in GrpcRecvTensorAsync()
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| D | mpi_rendezvous_mgr.cc | 216 Device* src_dev = nullptr; in AddRequest() local
218 parsed.src_device, &src_dev); in AddRequest()
241 if (src_dev->tensorflow_gpu_device_info() && in AddRequest()
245 val, src_dev, send_args.device_context, in AddRequest()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker.cc | 314 Device** src_dev) { in PrepareRecvTensor() argument
317 TF_RETURN_IF_ERROR(env_->device_mgr->LookupDevice(local_name, src_dev)); in PrepareRecvTensor()
320 if ((*src_dev)->attributes().incarnation() != parsed.src_incarnation) { in PrepareRecvTensor()
323 parsed.src_incarnation, " vs. ", (*src_dev)->attributes().incarnation(), in PrepareRecvTensor()
|
| D | worker.h | 96 Device** src_dev);
|
| /external/e2fsprogs/misc/ |
| D | create_inode.h | 13 dev_t src_dev; member
|
| D | create_inode.c | 599 if (hdlinks->hdl[i].src_dev == dev && in is_hardlink()
936 hdlinks->hdl[hdlinks->count].src_dev = st.st_dev; in __populate_fs()
|
| /external/autotest/server/ |
| D | site_linux_system.py | 477 self.clone_mac_address(src_dev=same_phy_as,
586 def clone_mac_address(self, src_dev=None, dst_dev=None): argument
595 interface.Interface(src_dev, self.host).mac_address)
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma.cc | 1020 const Rendezvous::ParsedKey& parsed, Device** src_dev) { in PrepareRecvTensor() argument
1024 local_name, src_dev)); in PrepareRecvTensor()
1027 if ((*src_dev)->attributes().incarnation() != parsed.src_incarnation) { in PrepareRecvTensor()
1030 parsed.src_incarnation, " vs. ", (*src_dev)->attributes().incarnation(), in PrepareRecvTensor()
|
| D | rdma.h | 347 Device** src_dev);
|