HomeSort by relevance Sort by last modified time
    Searched defs:dst_dev (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/verbs/
rdma_rendezvous_mgr.cc 69 Device* dst_dev; local
70 s = env_->device_mgr->LookupDevice(parsed.dst_device, &dst_dev);
78 rc->InsertTensorRequest(key, step_id_, dst_dev, recv_args, done);
  /external/tensorflow/tensorflow/contrib/nccl/kernels/
nccl_rewrite.cc 125 int dst_dev = edge->IsControlEdge() local
128 if (out_nodes_map.size() <= dst_dev) {
129 out_nodes_map.resize(dst_dev + 1);
130 recv_index_map.resize(dst_dev + 1);
132 auto it = out_nodes_map.begin() + dst_dev;
134 recv_index_map[dst_dev] = num_devices;

Completed in 57 milliseconds