Searched defs:sparsifier (Results 1 – 12 of 12) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ao/sparsity/ |
| H A D | test_data_scheduler.py | 21 def __init__(self, sparsifier, sparsifier_hyperparam, last_epoch=-1, verbose=False): argument
67 def _get_scheduler(self, sparsifier, schedule_param): argument
|
| H A D | test_scheduler.py | 110 def _get_sparsity_levels(sparsifier, precision=32): argument
|
| H A D | test_composability.py | 60 def _squash_mask_calibrate_and_convert(model, sparsifier, input): argument
|
| /aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/data_scheduler/ |
| H A D | README.md | 64 ### Note:
|
| /aosp_15_r20/external/pytorch/torch/ao/pruning/scheduler/ |
| H A D | lambda_scheduler.py | 33 def __init__(self, sparsifier, sl_lambda, last_epoch=-1, verbose=False): argument
|
| H A D | cubic_scheduler.py | 44 sparsifier, argument
|
| H A D | base_scheduler.py | 14 def __init__(self, sparsifier, last_epoch=-1, verbose=False): argument
|
| /aosp_15_r20/external/libopus/dnn/torch/lpcnet/utils/sparsification/ |
| H A D | gru_sparsifier.py | 184 sparsifier = GRUSparsifier([(gru, sparsify_dict)], 0, 100, 10) variable
|
| /aosp_15_r20/external/libopus/dnn/torch/dnntools/dnntools/sparsification/ |
| H A D | linear_sparsifier.py | 123 sparsifier = LinearSparsifier([(linear, params)], 0, 100, 5) variable
|
| H A D | conv1d_sparsifier.py | 128 sparsifier = Conv1dSparsifier([(conv, params)], 0, 100, 5) variable
|
| H A D | conv_transpose1d_sparsifier.py | 129 sparsifier = ConvTranspose1dSparsifier([(conv, params)], 0, 100, 5) variable
|
| H A D | gru_sparsifier.py | 173 sparsifier = GRUSparsifier([(gru, sparsify_dict)], 0, 100, 10) variable
|