Searched refs:RegistrationInfo (Results 1 – 2 of 2) sorted by relevance
30 struct RegistrationInfo { struct35 RegistrationInfo(const string& n, CollectiveRegistry::Factory f) in RegistrationInfo() argument44 std::vector<RegistrationInfo>* MutableCollectiveRegistry() { in MutableCollectiveRegistry()45 static std::vector<RegistrationInfo>* registry = in MutableCollectiveRegistry()46 new std::vector<RegistrationInfo>; in MutableCollectiveRegistry()207 std::vector<RegistrationInfo>* registry = MutableCollectiveRegistry(); in GetAll()208 for (const RegistrationInfo& reg_info : *registry) in GetAll()215 std::vector<RegistrationInfo>* registry = MutableCollectiveRegistry(); in Register()216 for (const RegistrationInfo& reg_info : *registry) { in Register()229 std::vector<RegistrationInfo>* registry = MutableCollectiveRegistry(); in LookupHelper()[all …]
33 struct RegistrationInfo { struct34 RegistrationInfo(DeviceType s, DeviceType r, CopyTensor::CopyFunction cf) in RegistrationInfo() argument45 std::vector<RegistrationInfo>* MutableRegistry() { in MutableRegistry()46 static std::vector<RegistrationInfo>* registry = in MutableRegistry()47 new std::vector<RegistrationInfo>; in MutableRegistry()230 std::vector<RegistrationInfo>* registry = MutableRegistry(); in ViaDMA()231 for (const RegistrationInfo& ri : *registry) { in ViaDMA()300 std::vector<RegistrationInfo>* registry = MutableRegistry(); in Register()