Searched refs:_test_addr_vs_numpy (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_linalg.py | 750 def _test_addr_vs_numpy(self, device, dtype, beta=1, alpha=1): member in TestLinalg
802 self._test_addr_vs_numpy(device, dtype, beta=True, alpha=False)
803 self._test_addr_vs_numpy(device, dtype, beta=False, alpha=True)
804 self._test_addr_vs_numpy(device, dtype, beta=False, alpha=False)
805 self._test_addr_vs_numpy(device, dtype, beta=True, alpha=True)
811 self._test_addr_vs_numpy(device, dtype, beta=2., alpha=1)
814 self._test_addr_vs_numpy(device, dtype, beta=2, alpha=1.)
817 self._test_addr_vs_numpy(device, dtype, beta=True, alpha=1)
820 self._test_addr_vs_numpy(device, dtype, beta=2, alpha=True)
823 self._test_addr_vs_numpy(device, dtype, beta=0, alpha=2)
[all …]
|