Searched refs:assert_and_get_unique_device (Results 1 – 4 of 4) sorted by relevance
11 assert_and_get_unique_device,
248 def assert_and_get_unique_device(module: torch.nn.Module) -> Any: function265 device = assert_and_get_unique_device(module)
56 assert_and_get_unique_device,822 device = assert_and_get_unique_device(float_module)
71 assert_and_get_unique_device,488 model_device = assert_and_get_unique_device(model)