Searched defs:sparsity_level (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/ |
| H A D | data_norm_sparsifier.py | 120 def __get_data_level_mask(self, data, sparsity_level, sparse_block_shape): argument
160 self, name, data, sparsity_level, sparse_block_shape, zeros_per_block, **kwargs argument
|
| /aosp_15_r20/external/pytorch/torch/ao/pruning/sparsifier/ |
| H A D | weight_norm_sparsifier.py | 106 self, data, input_shape, sparsity_level, sparse_block_shape, mask=None argument
215 sparsity_level, argument
|
| /aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/pruner/ |
| H A D | FPGM_pruner.py | 78 def update_mask(self, module, tensor_name, sparsity_level, **kwargs): argument
|
| /aosp_15_r20/external/pytorch/test/ao/sparsity/ |
| H A D | test_activation_sparsifier.py | 300 def _vanilla_norm_sparsifier(data, sparsity_level): argument
|