Home
last modified time | relevance | path

Searched defs:sparse_config (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/test/ao/sparsity/
Dtest_activation_sparsifier.py49 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/
Dquantization_utils.py32 **sparse_config,
/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/benchmarks/
Devaluate_disk_savings.py17 def create_attach_sparsifier(model, **sparse_config):
/external/pytorch/torch/ao/pruning/_experimental/activation_sparsifier/
Dactivation_sparsifier.py98 **sparse_config,
193 **sparse_config,