Lines Matching defs:amount
447 def __init__(self, amount): argument
473 def apply(cls, module, name, amount): argument
504 def __init__(self, amount): argument
532 def apply(cls, module, name, amount, importance_scores=None): argument
572 def __init__(self, amount, dim=-1): argument
642 def apply(cls, module, name, amount, dim=-1): argument
679 def __init__(self, amount, n, dim=-1): argument
757 def apply(cls, module, name, amount, n, dim, importance_scores=None): argument
854 def random_unstructured(module, name, amount): argument
891 def l1_unstructured(module, name, amount, importance_scores=None): argument
935 def random_structured(module, name, amount, dim): argument
976 def ln_structured(module, name, amount, n, dim, importance_scores=None): argument
1234 def _validate_pruning_amount_init(amount): argument
1264 def _validate_pruning_amount(amount, tensor_size): argument
1309 def _compute_nparams_toprune(amount, tensor_size): argument