Searched refs:new_backend (Results 1 – 3 of 3) sorted by relevance
188 BackendComponent new_backend = toBackendComponent(device.type()); in _change_backend_component_keys() local198 key_set = key_set | c10::getAutocastRelatedKeySetFromBackend(new_backend); in _change_backend_component_keys()203 key_set_ = key_set | DispatchKeySet(new_backend); in _change_backend_component_keys()
78 new_backend = OrtBackend.get_cached_instance_for_options(options)81 self.assertIs(reused_backend, new_backend)86 new_backend,
902 new_backend = "nccl"904 new_backend = "gloo"906 new_backend = "gloo"908 group, group_id, rank = initializer(backend=new_backend)912 if new_backend == "gloo":914 if new_backend == "nccl":