Home
last modified time | relevance | path

Searched defs:sparse_coo (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/
Dbase_data_sparsifier.py152 def _convert_mask(self, states, sparse_coo=True): argument
/external/pytorch/torch/ao/pruning/_experimental/activation_sparsifier/
Dactivation_sparsifier.py383 def _convert_mask(self, states_dict, sparse_coo=True): argument