assert_and_get_unique_device

class torch.ao.quantization.fx.utils.assert_and_get_unique_device(module) [source]

Возвращает уникальное устройство для модуля или None, если устройство не найдено. Вызывает ошибку, если обнаружено несколько устройств.