Searched refs:gpu_ring_order (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_param_resolver_local_test.cc | 63 const std::vector<int32>& gpu_ring_order, in RunCompleteDefaultRanking() argument
65 if (!gpu_ring_order.empty()) { in RunCompleteDefaultRanking()
66 group.gpu_ring_order = ""; in RunCompleteDefaultRanking()
67 for (int i = 0; i < static_cast<int32>(gpu_ring_order.size() - 1); ++i) { in RunCompleteDefaultRanking()
68 group.gpu_ring_order = in RunCompleteDefaultRanking()
69 strings::StrCat(group.gpu_ring_order, gpu_ring_order[i], ","); in RunCompleteDefaultRanking()
71 group.gpu_ring_order = in RunCompleteDefaultRanking()
72 strings::StrCat(group.gpu_ring_order, gpu_ring_order.back()); in RunCompleteDefaultRanking()
74 VLOG(2) << "gpu_ring_order " << group.gpu_ring_order; in RunCompleteDefaultRanking()
|
| D | base_collective_executor.h | 100 const DeviceMgr* dev_mgr, const string* gpu_ring_order, in BaseCollectiveExecutor() argument
106 gpu_ring_order_(gpu_ring_order), in BaseCollectiveExecutor()
|
| D | collective_param_resolver_local.cc | 143 gr->group.gpu_ring_order = cp->group.gpu_ring_order; in CompleteGroupLocal()
338 void OrderTaskDeviceMap(const string& gpu_ring_order, TaskDeviceMap* tdm) { in OrderTaskDeviceMap() argument
342 if (ParseRingOrder(gpu_ring_order, tdm)) return; in OrderTaskDeviceMap()
429 OrderTaskDeviceMap(gp.gpu_ring_order, &tdm); in EstablishGlobalRank()
|
| D | base_collective_executor.cc | 349 cp->group.gpu_ring_order = *gpu_ring_order_; in CompleteParamsAsync()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 76 string gpu_ring_order = ""; member
|