Searched refs:test_tensor_pow_tensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_binary_ufuncs.py | 1747 def test_tensor_pow_tensor(self, device): member in TestBinaryUfuncs
1751 def test_tensor_pow_tensor(values, torch_type, numpy_type): function
1759 test_tensor_pow_tensor(ints, torch.uint8, np.uint8)
1760 test_tensor_pow_tensor(ints, torch.int8, np.int8)
1761 test_tensor_pow_tensor(ints, torch.int16, np.int16)
1762 test_tensor_pow_tensor(ints, torch.int32, np.int32)
1763 test_tensor_pow_tensor(ints, torch.int64, np.int64)
1766 test_tensor_pow_tensor(floats, torch.float16, np.float16)
1767 test_tensor_pow_tensor(floats, torch.float32, np.float32)
1768 test_tensor_pow_tensor(floats, torch.float64, np.float64)
|