Searched defs:sparse_config (Results 1 – 4 of 4) 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
|
| /external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/ |
| D | quantization_utils.py | 32 **sparse_config,
|
| /external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/benchmarks/ |
| D | evaluate_disk_savings.py | 17 def create_attach_sparsifier(model, **sparse_config):
|
| /external/pytorch/torch/ao/pruning/_experimental/activation_sparsifier/ |
| D | activation_sparsifier.py | 98 **sparse_config,
193 **sparse_config,
|