Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.cc178 std::vector<int32> gpu_ring_order_vec; in ParseRingOrder() local
180 &gpu_ring_order_vec)) { in ParseRingOrder()
183 if (gpu_ring_order_vec.size() != tdm->size()) return false; in ParseRingOrder()
186 for (int32 rank = 0; rank < static_cast<int32>(gpu_ring_order_vec.size()); in ParseRingOrder()
188 gpu_ranks[gpu_ring_order_vec[rank]] = rank; in ParseRingOrder()