Searched refs:_test_multinomial_empty (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/pytorch/test/ |
H A D | test_torch.py | 5629 def _test_multinomial_empty(self, device, replacement, num_samples): member in TestTorchDeviceType 5637 self._test_multinomial_empty(device, True, 1) 5638 self._test_multinomial_empty(device, True, 2) 5642 self._test_multinomial_empty(device, False, 1) 5643 self._test_multinomial_empty(device, False, 2)
|