Searched defs:except_for (Results 1 – 4 of 4) sorted by relevance
| /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
879 except_for=None, argument
|
| H A D | jit_utils.py | 158 def assertAllFused(self, graph, except_for=()): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit_fuser.py | 43 def assertAllFused(self, graph, except_for=()): argument
|
| H A D | test_jit_fuser_te.py | 156 def assertAllFused(self, graph, except_for=None): argument
|