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 | 539 dst_dev=net_dev.if_name) 647 def clone_mac_address(self, src_dev=None, dst_dev=None): argument 655 dst_dev,
|