Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/lightning/callbacks/
H A D_data_sparstity_utils.py12 def _attach_model_to_data_sparsifier(module, data_sparsifier, config=None): argument
36 def _log_sparsified_level(model, data_sparsifier) -> None: argument
/aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/data_scheduler/
H A Dbase_data_scheduler.py34 self, data_sparsifier, schedule_param: str, last_epoch=-1, verbose=False argument
/aosp_15_r20/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/lightning/tests/
H A Dtest_callbacks.py61 data_sparsifier, argument