Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_param_resolver_local.cc155 typedef std::unordered_map<string, DevRec> TaskDeviceMap; typedef
156 typedef std::unordered_map<string, TaskDeviceMap> GlobalDeviceMap;
165 TaskDeviceMap& tdm = gdm[ip.task_names[i]]; in BuildDevRecs()
177 bool ParseRingOrder(const string& gpu_ring_order_str, TaskDeviceMap* tdm) { in ParseRingOrder()
205 void OrderTaskDeviceMap(const string& gpu_ring_order, TaskDeviceMap* tdm) { in OrderTaskDeviceMap()
297 TaskDeviceMap& tdm = iter.second; in EstablishGlobalRank()
309 TaskDeviceMap* tdm = &gdm[task_name]; in EstablishGlobalRank()
471 const TaskDeviceMap& tdm = git.second; in CompleteDefaultRanking()