Home
last modified time | relevance | path

Searched defs:squash_mask (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/ao/pruning/sparsifier/
H A Dbase_sparsifier.py220 def squash_mask( member in BaseSparsifier
/aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/
H A Dbase_data_sparsifier.py287 def squash_mask(self, *args, leave_parametrized=True, names=None, **kwargs): member in BaseDataSparsifier
/aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/activation_sparsifier/
H A Dactivation_sparsifier.py349 def squash_mask(self, attach_sparsify_hook=True, **kwargs): member in ActivationSparsifier