Searched refs:deviceTypeToBackend_ (Results 1 – 2 of 2) sorted by relevance
91 if (deviceTypeToBackend_.find(deviceType) != deviceTypeToBackend_.end()) { in getBackend()92 return deviceTypeToBackend_.at(deviceType); in getBackend()107 deviceTypeToBackend_[deviceType] = backend; in getBackend()144 TORCH_CHECK(!deviceTypeToBackend_.empty(), "ProcessGroup name not set"); in getGroupName()145 return deviceTypeToBackend_.begin()->second->getGroupUid(); in getGroupName()149 for (auto& kv : deviceTypeToBackend_) { in setGroupName()161 for (auto& kv : deviceTypeToBackend_) { in setGroupDesc()167 for (auto& kv : deviceTypeToBackend_) { in enableCollectivesTiming()174 deviceTypeToBackend_.clear(); in release_resources()
631 deviceTypeToBackend_[deviceType] = existingBackend; in setBackend()638 deviceTypeToBackend_[deviceType] = backend.value(); in setBackend()741 deviceTypeToBackend_; member in c10d::ProcessGroup