Searched defs:only_for (Results 1 – 11 of 11) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_functionalization_of_rng_ops.py | 323 only_for = ("cuda",) variable
343 only_for = ("cpu",) variable
|
| H A D | test_maskedtensor.py | 1005 only_for = ("cpu", "cuda") variable
|
| H A D | test_jit_fuser_te.py | 2973 only_for = ("cuda") if IS_FBCODE else ("cpu", "cuda") variable
|
| H A D | test_proxy_tensor.py | 2169 only_for = ("cpu") variable
|
| H A D | test_custom_ops.py | 3934 only_for = ("cpu", "cuda") variable
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_device_type.py | 741 def filter_desired_device_types(device_type_test_bases, except_for=None, only_for=None): argument
808 except_for=None, only_for=None, include_lazy=False, allow_mps=False, allow_xpu=False argument
880 only_for=None, argument
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_ops.py | 2992 only_for = ("cpu", "cuda") variable
|
| H A D | test_vmap.py | 6404 only_for = ("cpu", "cuda") variable
|
| H A D | test_aotdispatch.py | 6332 only_for = "cpu" variable
|
| H A D | test_eager_transforms.py | 5200 only_for = ("cpu", "cuda") variable
|
| /aosp_15_r20/external/pytorch/test/nn/ |
| H A D | test_parametrization.py | 1900 only_for = ("cpu", "cuda") variable
|