Searched refs:_test_constructor_on_device (Results 1 – 1 of 1) sorted by relevance
192 def _test_constructor_on_device(self, model, device): member in TestBaseStructuredSparsifier219 self._test_constructor_on_device(model, torch.device(device))