| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_spectral_ops.py | 16 (instantiate_device_type_tests, ops, dtypes, onlyNativeDeviceTypes,
131 @onlyNativeDeviceTypes
173 @onlyNativeDeviceTypes
234 @onlyNativeDeviceTypes
243 @onlyNativeDeviceTypes
253 @onlyNativeDeviceTypes
267 @onlyNativeDeviceTypes
319 @onlyNativeDeviceTypes
336 @onlyNativeDeviceTypes
353 @onlyNativeDeviceTypes
[all …]
|
| H A D | test_dlpack.py | 9 onlyNativeDeviceTypes,
22 @onlyNativeDeviceTypes
39 @onlyNativeDeviceTypes
56 @onlyNativeDeviceTypes
87 @onlyNativeDeviceTypes
104 @onlyNativeDeviceTypes
156 @onlyNativeDeviceTypes
215 @onlyNativeDeviceTypes
|
| H A D | test_view_ops.py | 15 onlyNativeDeviceTypes,
155 @onlyNativeDeviceTypes
269 @onlyNativeDeviceTypes
305 @onlyNativeDeviceTypes
370 @onlyNativeDeviceTypes
396 @onlyNativeDeviceTypes
407 @onlyNativeDeviceTypes
417 @onlyNativeDeviceTypes
427 @onlyNativeDeviceTypes
437 @onlyNativeDeviceTypes
[all …]
|
| H A D | test_type_promotion.py | 13 ….testing._internal.common_device_type import (instantiate_device_type_tests, onlyNativeDeviceTypes,
288 @onlyNativeDeviceTypes
631 @onlyNativeDeviceTypes
742 @onlyNativeDeviceTypes
900 @onlyNativeDeviceTypes
905 @onlyNativeDeviceTypes
910 @onlyNativeDeviceTypes
916 @onlyNativeDeviceTypes
921 @onlyNativeDeviceTypes
932 @onlyNativeDeviceTypes
[all …]
|
| H A D | test_decomp.py | 22 onlyNativeDeviceTypes,
548 @onlyNativeDeviceTypes
557 @onlyNativeDeviceTypes
574 @onlyNativeDeviceTypes
1041 @onlyNativeDeviceTypes
1056 @onlyNativeDeviceTypes
1122 @onlyNativeDeviceTypes
1135 @onlyNativeDeviceTypes
1146 @onlyNativeDeviceTypes
|
| H A D | test_tensor_creation_ops.py | 23 expectedFailureMeta, instantiate_device_type_tests, deviceCountAtLeast, onlyNativeDeviceTypes,
102 @onlyNativeDeviceTypes
114 @onlyNativeDeviceTypes
412 @onlyNativeDeviceTypes
421 @onlyNativeDeviceTypes
434 @onlyNativeDeviceTypes
446 @onlyNativeDeviceTypes
462 @onlyNativeDeviceTypes
618 @onlyNativeDeviceTypes
640 @onlyNativeDeviceTypes
[all …]
|
| H A D | test_torch.py | 54 skipMeta, PYTORCH_CUDA_MEMCHECK, largeTensorTest, onlyNativeDeviceTypes,
160 @onlyNativeDeviceTypes
206 @onlyNativeDeviceTypes
241 @onlyNativeDeviceTypes
251 @onlyNativeDeviceTypes
268 @onlyNativeDeviceTypes
306 @onlyNativeDeviceTypes
320 @onlyNativeDeviceTypes
333 @onlyNativeDeviceTypes
914 @onlyNativeDeviceTypes
[all …]
|
| H A D | test_masked.py | 19 (instantiate_device_type_tests, ops, onlyNativeDeviceTypes, precisionOverride)
274 @onlyNativeDeviceTypes
296 @onlyNativeDeviceTypes
|
| H A D | test_binary_ufuncs.py | 27 onlyNativeDeviceTypes,
1153 @onlyNativeDeviceTypes
1191 @onlyNativeDeviceTypes
1688 @onlyNativeDeviceTypes
1719 @onlyNativeDeviceTypes
1889 @onlyNativeDeviceTypes
1962 @onlyNativeDeviceTypes
2721 @onlyNativeDeviceTypes # Check Issue https://github.com/pytorch/pytorch/issues/48130
2874 @onlyNativeDeviceTypes
2899 @onlyNativeDeviceTypes
[all …]
|
| H A D | test_shape_ops.py | 21 onlyNativeDeviceTypes,
271 @onlyNativeDeviceTypes
766 @onlyNativeDeviceTypes
|
| H A D | test_unary_ufuncs.py | 38 onlyNativeDeviceTypes,
1252 @onlyNativeDeviceTypes
1264 @onlyNativeDeviceTypes
1275 @onlyNativeDeviceTypes
1304 @onlyNativeDeviceTypes
|
| H A D | test_indexing.py | 20 onlyNativeDeviceTypes,
180 @onlyNativeDeviceTypes
944 @onlyNativeDeviceTypes
1031 @onlyNativeDeviceTypes
1051 @onlyNativeDeviceTypes
|
| H A D | test_nn.py | 45 …onlyNativeDeviceTypes, deviceCountAtLeast, largeTensorTest, expectedFailureMeta, expectedFailureMP…
8345 @onlyNativeDeviceTypes
8393 @onlyNativeDeviceTypes
8467 @onlyNativeDeviceTypes
8493 @onlyNativeDeviceTypes
8523 @onlyNativeDeviceTypes
8554 @onlyNativeDeviceTypes
8677 @onlyNativeDeviceTypes
8693 @onlyNativeDeviceTypes
8721 @onlyNativeDeviceTypes
[all …]
|
| H A D | test_linalg.py | 26 skipCUDAIfNoMagmaAndNoCusolver, skipCUDAIfRocm, onlyNativeDeviceTypes, dtypesIfCUDA,
2852 @onlyNativeDeviceTypes # TODO: XLA doesn't raise exception
2873 @onlyNativeDeviceTypes # TODO: XLA doesn't raise exception
3358 @onlyNativeDeviceTypes
3363 @onlyNativeDeviceTypes
3544 @onlyNativeDeviceTypes
3559 @onlyNativeDeviceTypes
3594 @onlyNativeDeviceTypes
6407 @onlyNativeDeviceTypes
6428 @onlyNativeDeviceTypes
[all …]
|
| H A D | test_reductions.py | 26 onlyNativeDeviceTypes, onlyCUDA, largeTensorTest, ops, precisionOverride)
1049 @onlyNativeDeviceTypes
1224 @onlyNativeDeviceTypes
1238 @onlyNativeDeviceTypes
1765 @onlyNativeDeviceTypes
1772 @onlyNativeDeviceTypes
1997 @onlyNativeDeviceTypes
2082 @onlyNativeDeviceTypes
2581 @onlyNativeDeviceTypes
|
| H A D | test_sort_and_select.py | 19 onlyNativeDeviceTypes,
885 @onlyNativeDeviceTypes
1140 @onlyNativeDeviceTypes # Fails on XLA
|
| H A D | test_testing.py | 25 …et_device_type_test_bases, instantiate_device_type_tests, onlyCPU, onlyCUDA, onlyNativeDeviceTypes,
129 @onlyNativeDeviceTypes
399 @onlyNativeDeviceTypes
|
| H A D | test_sparse.py | 23 deviceCountAtLeast, OpDTypes, onlyNativeDeviceTypes)
4982 @onlyNativeDeviceTypes
5003 @onlyNativeDeviceTypes
5026 @onlyNativeDeviceTypes
5071 @onlyNativeDeviceTypes
5117 @onlyNativeDeviceTypes
5172 @onlyNativeDeviceTypes
5229 @onlyNativeDeviceTypes
|
| H A D | test_optim.py | 28 onlyNativeDeviceTypes,
1067 @onlyNativeDeviceTypes
|
| H A D | test_cuda.py | 41 onlyNativeDeviceTypes,
4621 @onlyNativeDeviceTypes
|
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_pooling.py | 28 onlyNativeDeviceTypes,
496 @onlyNativeDeviceTypes
515 @onlyNativeDeviceTypes
548 @onlyNativeDeviceTypes
558 @onlyNativeDeviceTypes
568 @onlyNativeDeviceTypes
575 @onlyNativeDeviceTypes
582 @onlyNativeDeviceTypes
596 @onlyNativeDeviceTypes
610 @onlyNativeDeviceTypes
[all …]
|
| H A D | test_convolution.py | 29 onlyNativeDeviceTypes,
2131 decorators=[onlyNativeDeviceTypes, disableMkldnn, disablecuDNN],
2143 decorators=[onlyNativeDeviceTypes, disableMkldnn, disablecuDNN],
2155 decorators=[onlyNativeDeviceTypes, disableMkldnn, disablecuDNN],
2167 decorators=[onlyNativeDeviceTypes, disableMkldnn, disablecuDNN],
2179 decorators=[onlyNativeDeviceTypes, disableMkldnn, disablecuDNN],
2191 decorators=[onlyNativeDeviceTypes, disableMkldnn, disablecuDNN],
2203 decorators=[onlyNativeDeviceTypes, disableMkldnn, disablecuDNN],
2215 decorators=[onlyNativeDeviceTypes, disableMkldnn, disablecuDNN],
2255 decorators=[onlyNativeDeviceTypes, disableMkldnn, disablecuDNN],
[all …]
|
| H A D | test_embedding.py | 17 onlyNativeDeviceTypes,
488 @onlyNativeDeviceTypes
638 @onlyNativeDeviceTypes
1589 @onlyNativeDeviceTypes # currently fails on XLA
1612 @onlyNativeDeviceTypes # currently fails on XLA
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_torchinductor_opinfo.py | 24 onlyNativeDeviceTypes,
566 @onlyNativeDeviceTypes
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_device_type.py | 1402 def onlyNativeDeviceTypes(fn): function
|