Searched defs:param_kwargs (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_device_type.py | 318 def _update_param_kwargs(param_kwargs, name, value): argument
413 def _apply_precision_override_for_test(self, test, param_kwargs): argument
424 cls, name, *, test, param_kwargs=None, decorator_fn=lambda _: [] argument
441 def instantiated_test(self, param_kwargs=param_kwargs): argument
|
| H A D | common_optimizers.py | 176 def get_decorators(self, test_class, test_name, device, dtype, param_kwargs): argument
|
| H A D | common_utils.py | 494 def merged_decorator_fn(param_kwargs, old_dec_fn=old_dec_fn, new_dec_fn=new_dec_fn): argument
529 def instantiate_test_helper(cls, name, test, param_kwargs): argument
531 def instantiated_test(self, param_kwargs=param_kwargs): argument
|
| H A D | common_modules.py | 240 def get_decorators(self, test_class, test_name, device, dtype, param_kwargs): argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/ |
| H A D | core.py | 105 def is_active(self, cls_name, test_name, device_type, dtype, param_kwargs): argument
1393 def get_decorators(self, test_class, test_name, device, dtype, param_kwargs): argument
|