Searched refs:_test_step_linear_on_device (Results 1 – 1 of 1) sorted by relevance
270 def _test_step_linear_on_device(self, model, device): member in TestBaseStructuredSparsifier289 self._test_step_linear_on_device(model, torch.device(device))