Home
last modified time | relevance | path

Searched refs:get_dlrm_model (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/benchmarks/
Ddlrm_utils.py43 def get_dlrm_model(sparse_dlrm=False): function
169 model = get_dlrm_model(sparse_dlrm=sparse_dlrm)
Devaluate_disk_savings.py11 from dlrm_utils import get_dlrm_model, get_valid_name # type: ignore[import]
115 orig_model = get_dlrm_model()