Searched refs:_assert_and_get_unique_device (Results 1 – 3 of 3) sorted by relevance
6 from torch.ao.quantization.utils import _assert_and_get_unique_device141 device = _assert_and_get_unique_device(m)
30 _assert_and_get_unique_device,253 return _assert_and_get_unique_device(module)
762 def _assert_and_get_unique_device(module: torch.nn.Module) -> Any: function