HomeSort by relevance Sort by last modified time
    Searched refs:gpu_ring_order (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/
collective_param_resolver_local_test.cc 51 const std::vector<int32>& gpu_ring_order,
61 if (!gpu_ring_order.empty()) {
62 ir.shared.instance.gpu_ring_order = "";
63 for (int i = 0; i < static_cast<int32>(gpu_ring_order.size() - 1);
65 ir.shared.instance.gpu_ring_order = strings::StrCat(
66 ir.shared.instance.gpu_ring_order, gpu_ring_order[i], ",");
68 ir.shared.instance.gpu_ring_order = strings::StrCat(
69 ir.shared.instance.gpu_ring_order, gpu_ring_order.back())
    [all...]
base_collective_executor.h 99 const string* gpu_ring_order)
104 gpu_ring_order_(gpu_ring_order) {}
base_collective_executor.cc 279 cp->instance.gpu_ring_order = *gpu_ring_order_;
collective_param_resolver_local.cc 205 void OrderTaskDeviceMap(const string& gpu_ring_order, TaskDeviceMap* tdm) {
209 if (ParseRingOrder(gpu_ring_order, tdm)) return;
298 OrderTaskDeviceMap(cp->instance.gpu_ring_order, &tdm);
  /external/tensorflow/tensorflow/core/framework/
collective.cc 69 gpu_ring_order = other.gpu_ring_order;
collective.h 95 string gpu_ring_order = ""; member in struct:tensorflow::CollInstanceParams

Completed in 113 milliseconds