Searched refs:dst_dev (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/nccl/kernels/ |
| D | nccl_rewrite.cc | 125 int dst_dev = edge->IsControlEdge() in ReplaceBroadcast() local
128 if (out_nodes_map.size() <= dst_dev) { in ReplaceBroadcast()
129 out_nodes_map.resize(dst_dev + 1); in ReplaceBroadcast()
130 recv_index_map.resize(dst_dev + 1); in ReplaceBroadcast()
132 auto it = out_nodes_map.begin() + dst_dev; in ReplaceBroadcast()
134 recv_index_map[dst_dev] = num_devices; in ReplaceBroadcast()
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma_rendezvous_mgr.cc | 69 Device* dst_dev; in RecvFromRemoteAsync() local
70 s = env_->device_mgr->LookupDevice(parsed.dst_device, &dst_dev); in RecvFromRemoteAsync()
78 rc->InsertTensorRequest(key, step_id_, dst_dev, recv_args, done); in RecvFromRemoteAsync()
|
| D | rdma.h | 240 RdmaChannel* channel, Device* dst_dev,
432 const string& key, int64 step_id, Device* dst_dev,
|
| D | rdma.cc | 669 const string& key, int64 step_id, Device* dst_dev, in InsertTensorRequest() argument
677 RdmaTensorRequest request(request_index, key, step_id, this, dst_dev, in InsertTensorRequest()
1476 Device* dst_dev, const Rendezvous::Args recv_args, in RdmaTensorRequest() argument
1482 dst_dev_(dst_dev), in RdmaTensorRequest()
|
| /external/autotest/server/ |
| D | site_linux_system.py | 478 dst_dev=net_dev.if_name)
586 def clone_mac_address(self, src_dev=None, dst_dev=None): argument
594 dst_dev,
|