Searched defs:data_with_config (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ao/sparsity/ |
| H A D | test_data_sparsifier.py | 47 def run_all_checks(self, data_list, data_with_config, defaults): argument
73 data_with_config, argument
95 def check_constructor(self, data_list, data_with_config, defaults, **kwargs): argument
114 def check_step(self, data_list, data_with_config, defaults, **kwargs): argument
147 def check_squash_mask(self, data_list, data_with_config, defaults, **kwargs): argument
167 def check_add_data(self, data_list, data_with_config, defaults, **kwargs): argument
198 def check_state_dict(self, data_list, data_with_config, defaults, **kwargs): argument
247 def check_memory_reference(self, data_list, data_with_config, defaults, **kwargs): argument
280 def run_all_checks(self, data_list, defaults, data_with_config, norm_type="L1"): argument
331 def check_step(self, data_list, data_with_config, defaults, norm_type="L1"): argument
[all …]
|
| H A D | test_data_scheduler.py | 56 def _get_sparsifier(self, data_list, data_with_config, defaults): argument
|