Home
last modified time | relevance | path

Searched refs:_test_multinomial_empty (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_torch.py5629 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)