Home
last modified time | relevance | path

Searched refs:device_cls (Results 1 – 6 of 6) sorted by relevance

/external/pytorch/test/
Dtest_testing.py1784 device_cls = locals()[f'TestParametrized{device.upper()}']
1785 expected_test_names = [name.format(device_cls.__name__, device) for name in (
1790 test_names = _get_test_names_for_test_class(device_cls)
1808 device_cls = locals()[f'TestParametrized{device.upper()}']
1809 expected_test_names = [name.format(device_cls.__name__, device) for name in (
1813 test_names = _get_test_names_for_test_class(device_cls)
1851 device_cls = locals()[f'TestParametrized{device.upper()}']
1852 expected_test_names = [name.format(device_cls.__name__, device) for name in (
1862 test_names = _get_test_names_for_test_class(device_cls)
1879 device_cls = locals()[f'TestParametrized{device.upper()}']
[all …]
Dtest_masked.py213 def _parametrize_test(self, test, generic_cls, device_cls): argument
/external/pytorch/torch/testing/_internal/
Dcommon_device_type.py478 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
1065 if device_cls is None:
1080 op.supported_backward_dtypes(device_cls.device_type)
1083 dtypes = op.supported_backward_dtypes(device_cls.device_type)
1086 op.supported_dtypes(device_cls.device_type)
1089 dtypes = set(op.supported_dtypes(device_cls.device_type))
1092 supported = op.supported_dtypes(device_cls.device_type)
1094 device_cls.device_type
[all …]
Dcommon_optimizers.py204 def _parametrize_test(self, test, generic_cls, device_cls): argument
205 if device_cls is None:
230 device_cls.device_type,
Dcommon_utils.py428 def _parametrize_test(self, test, generic_cls, device_cls): argument
477 def composite_fn(test, generic_cls, device_cls, argument
480 old_tests = list(old_parametrize_fn(test, generic_cls, device_cls))
483 new_parametrize_fn(old_test, generic_cls, device_cls):
538 class_attr, generic_cls=generic_cls, device_cls=None):
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
Dcommon_modules.py101 def _parametrize_test(self, test, generic_cls, device_cls): argument
102 if device_cls is None:
108 dtypes = set(module_info.supported_dtypes(device_cls.device_type))
135 test.__name__, device_cls.device_type, dtype)