Home
last modified time | relevance | path

Searched refs:dst_dev (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc127 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/
Dcollective_rma_local.cc120 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()
Dcollective_rma_local.h68 Device* dst_dev, const AllocatorAttributes& src_attr,
/external/autotest/server/
Dsite_linux_system.py539 dst_dev=net_dev.if_name)
647 def clone_mac_address(self, src_dev=None, dst_dev=None): argument
655 dst_dev,