Searched refs:sparse_block_shape (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/ |
| H A D | data_norm_sparsifier.py | 43 sparse_block_shape: Tuple[int, int] = (1, 4),
48 zeros_per_block = reduce(operator.mul, sparse_block_shape)
54 "sparse_block_shape": sparse_block_shape,
61 self, data, dim, indices, output_size, sparse_block_shape argument
68 kernel_size=sparse_block_shape,
69 stride=sparse_block_shape,
74 def __get_block_level_mask(self, data, sparse_block_shape, zeros_per_block): argument
77 block_height, block_width = sparse_block_shape
98 kernel_size=sparse_block_shape,
99 stride=sparse_block_shape,
[all …]
|
| /aosp_15_r20/external/pytorch/torch/ao/pruning/sparsifier/ |
| H A D | weight_norm_sparsifier.py | 62 sparse_block_shape: Tuple[int, int] = (1, 4),
67 zeros_per_block = reduce(operator.mul, sparse_block_shape)
70 "sparse_block_shape": sparse_block_shape,
106 self, data, input_shape, sparsity_level, sparse_block_shape, mask=None argument
117 block_h, block_w = sparse_block_shape
131 values_per_block = reduce(operator.mul, sparse_block_shape)
136 kernel_size=sparse_block_shape,
137 stride=sparse_block_shape,
155 block_shape=sparse_block_shape,
161 def _make_block_mask(self, data, sparse_block_shape, zeros_per_block, mask=None): argument
[all …]
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/sparse/quantized/ |
| H A D | linear.py | 235 …sparse_block_shape = mod.sparse_params.get("sparse_block_shape", None) # type: ignore[operator, u…
236 assert isinstance(sparse_block_shape, (tuple, list))
237 assert len(sparse_block_shape) == 2
|
| /aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/benchmarks/ |
| H A D | evaluate_disk_savings.py | 41 sparse_block_shape, argument
137 model, sparse_block_shape=sbs, norm=norm, sparsity_level=sl
|
| /aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/lightning/callbacks/ |
| H A D | README.md | 25 'sparse_block_shape': (1, 4),
43 'sparse_block_shape': (1, 4),
|
| /aosp_15_r20/external/pytorch/test/ao/sparsity/ |
| H A D | test_data_sparsifier.py | 307 sparse_block_shape = config["sparse_block_shape"]
310 block_height, block_width = sparse_block_shape
365 sparse_block_shape=(1, 4),
|
| H A D | test_sparsifier.py | 271 sparsity_level=1.0, sparse_block_shape=(1, 4), zeros_per_block=2
|
| /aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/pt/ |
| H A D | ao_sparsifier_test.py | 38 sparse_block_shape=SBS,
|