Searched defs:activation_sparsifier (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/test/ao/sparsity/ |
| D | test_activation_sparsifier.py | 49 def _check_constructor(self, activation_sparsifier, model, defaults, sparse_config): argument
64 self, activation_sparsifier, defaults, sparse_config, layer_args_list argument
110 def _check_pre_forward_hook(self, activation_sparsifier, data_list): argument
142 def _check_step(self, activation_sparsifier, data_agg_actual): argument
171 def _check_squash_mask(self, activation_sparsifier, data): argument
|