Searched refs:dst_dev (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_rewrite.cc | 127 int dst_dev = edge->IsControlEdge() in ReplaceBroadcast() local
130 if (out_nodes_map.size() <= dst_dev) { in ReplaceBroadcast()
131 out_nodes_map.resize(dst_dev + 1); in ReplaceBroadcast()
132 recv_index_map.resize(dst_dev + 1); in ReplaceBroadcast()
134 auto it = out_nodes_map.begin() + dst_dev; in ReplaceBroadcast()
136 recv_index_map[dst_dev] = num_devices; in ReplaceBroadcast()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_rma_local.cc | 120 Device* dst_dev, const AllocatorAttributes& src_attr, in MemCpyAsync() argument
132 dst_attr.on_host() ? DEVICE_CPU : dst_dev->attributes().device_type()); in MemCpyAsync()
154 src_dev_ctx, dst_dev_ctx, src_dev, dst_dev, src_attr, in MemCpyAsync()
|
| D | collective_rma_local.h | 68 Device* dst_dev, const AllocatorAttributes& src_attr,
|
| /external/autotest/server/ |
| D | site_linux_system.py | 537 dst_dev=net_dev.if_name)
645 def clone_mac_address(self, src_dev=None, dst_dev=None): argument
653 dst_dev,
|