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