Searched refs:_test_int_and_float_pow (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_binary_ufuncs.py | 1503 def _test_int_and_float_pow(dt, low, high, dev): function
1570 _test_int_and_float_pow(torch.int8, -2, 2, device)
1571 _test_int_and_float_pow(torch.uint8, 0, 3, device)
1572 _test_int_and_float_pow(torch.int16, -5, 5, device)
1573 _test_int_and_float_pow(torch.int64, -10, 10, device)
1574 _test_int_and_float_pow(torch.int32, -10, 10, device)
1575 _test_int_and_float_pow(torch.float16, 0.0, 5.0, device)
1576 _test_int_and_float_pow(torch.float32, 0.0, 10.0, device)
1577 _test_int_and_float_pow(torch.float64, 0.0, 10.0, device)
1579 _test_int_and_float_pow(torch.float32, -10.0, 10.0, device)
[all …]
|