assert_and_get_unique_device

class torch.ao.quantization.fx.utils.assert_and_get_unique_device(module)

Returns the unique device for a module, or None if no device is found. Throws an error if multiple devices are detected.

Return type:

Any