Searched defs:sparse_block_shape (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/ |
| H A D | data_norm_sparsifier.py | 61 self, data, dim, indices, output_size, sparse_block_shape argument
74 def __get_block_level_mask(self, data, sparse_block_shape, zeros_per_block): argument
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
161 def _make_block_mask(self, data, sparse_block_shape, zeros_per_block, mask=None): argument
216 sparse_block_shape, argument
|
| /aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/benchmarks/ |
| H A D | evaluate_disk_savings.py | 41 sparse_block_shape, argument
|