Searched refs:RegistrationInfo (Results 1 – 2 of 2) sorted by relevance
29 struct RegistrationInfo { struct34 RegistrationInfo(const string& n, CollectiveRegistry::Factory f) in RegistrationInfo() argument43 std::vector<RegistrationInfo>* MutableCollectiveRegistry() { in MutableCollectiveRegistry()44 static std::vector<RegistrationInfo>* registry = in MutableCollectiveRegistry()45 new std::vector<RegistrationInfo>; in MutableCollectiveRegistry()195 std::vector<RegistrationInfo>* registry = MutableCollectiveRegistry(); in GetAll()196 for (const RegistrationInfo& reg_info : *registry) in GetAll()203 std::vector<RegistrationInfo>* registry = MutableCollectiveRegistry(); in Register()204 for (const RegistrationInfo& reg_info : *registry) { in Register()217 std::vector<RegistrationInfo>* registry = MutableCollectiveRegistry(); in LookupHelper()[all …]
32 struct RegistrationInfo { struct33 RegistrationInfo(DeviceType s, DeviceType r, CopyTensor::CopyFunction cf) in RegistrationInfo() function44 std::vector<RegistrationInfo>* MutableRegistry() { in MutableRegistry()45 static std::vector<RegistrationInfo>* registry = in MutableRegistry()46 new std::vector<RegistrationInfo>; in MutableRegistry()277 std::vector<RegistrationInfo>* registry = MutableRegistry(); in ViaDMA()278 for (const RegistrationInfo& ri : *registry) { in ViaDMA()351 std::vector<RegistrationInfo>* registry = MutableRegistry(); in Register()