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)
|