| /external/tensorflow/tensorflow/contrib/gdr/ |
| gdr_worker.cc | 64 Device* src_dev = nullptr; local
66 s = PrepareRecvTensor(parsed, &src_dev);
81 [this, opts, response, done, src_dev, dma_ok](
93 (src_dev->tensorflow_gpu_device_info() == nullptr) ||
106 transport_options, val, src_dev, send_args.device_context,
119 if (src_dev->tensorflow_gpu_device_info() && (!on_host)) {
125 Allocator* alloc = src_dev->GetAllocator(alloc_attrs);
128 << "send dev name: " << src_dev->name()
129 << " gpu_info: " << src_dev->tensorflow_gpu_device_info();
139 GPUUtil::CopyGPUTensorToCPU(src_dev, send_dev_context, &val, copy
[all...] |
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| grpc_worker_service.cc | 379 Device* src_dev = nullptr; local
381 s = PrepareRecvTensor(parsed, &src_dev);
395 [opts, response, done, src_dev](const Status& status,
410 if (src_dev->tensorflow_gpu_device_info() && (!on_host)) {
416 Allocator* alloc = src_dev->GetAllocator(alloc_attrs);
419 << "send dev name: " << src_dev->name()
420 << " gpu_info: " << src_dev->tensorflow_gpu_device_info();
430 GPUUtil::CopyGPUTensorToCPU(src_dev, send_dev_context, &val, copy,
|
| /external/e2fsprogs/misc/ |
| create_inode.h | 13 dev_t src_dev; member in struct:hdlink_s
|
| create_inode.c | 599 if (hdlinks->hdl[i].src_dev == dev &&
936 hdlinks->hdl[hdlinks->count].src_dev = st.st_dev;
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| mpi_rendezvous_mgr.cc | 187 // if(src_dev->tensorflow_gpu_device_info()) //memcpy to send_buffer2_
216 Device* src_dev = nullptr;
218 parsed.src_device, &src_dev);
241 if (src_dev->tensorflow_gpu_device_info() &&
245 val, src_dev, send_args.device_context,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| worker.h | 96 Device** src_dev);
|
| worker.cc | 312 // device in "*src_dev".
314 Device** src_dev) {
317 TF_RETURN_IF_ERROR(env_->device_mgr->LookupDevice(local_name, src_dev));
320 if ((*src_dev)->attributes().incarnation() != parsed.src_incarnation) {
323 parsed.src_incarnation, " vs. ", (*src_dev)->attributes().incarnation(),
|
| /external/autotest/server/ |
| 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):
589 @param src_dev string name of device to copy address from.
595 interface.Interface(src_dev, self.host).mac_address)
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| rdma.cc | [all...] |
| rdma.h | 347 Device** src_dev);
|
