Home
last modified time | relevance | path

Searched defs:generic_cls (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_device_type.py422 def instantiate_test(cls, name, test, *, generic_cls=None): argument
478 def default_parametrize_fn(test, generic_cls, device_cls): argument
489 def dtype_parametrize_fn(test, generic_cls, device_cls, dtypes=dtypes): argument
1063 def _parametrize_test(self, test, generic_cls, device_cls): argument
H A Dcommon_utils.py428 def _parametrize_test(self, test, generic_cls, device_cls): argument
477 def composite_fn(test, generic_cls, device_cls, argument
502 def instantiate_parametrized_tests(generic_cls): argument
656 def _parametrize_test(self, test, generic_cls, device_cls): argument
749 def _parametrize_test(self, test, generic_cls, device_cls): argument
2020 def _parametrize_test(self, test, generic_cls, device_cls): argument
H A Dcommon_optimizers.py204 def _parametrize_test(self, test, generic_cls, device_cls): argument
H A Dcommon_modules.py101 def _parametrize_test(self, test, generic_cls, device_cls): argument
/aosp_15_r20/external/pytorch/test/
H A Dtest_masked.py213 def _parametrize_test(self, test, generic_cls, device_cls): argument