Searched defs:sparse_dlrm (Results 1 – 2 of 2) sorted by relevance
43 def get_dlrm_model(sparse_dlrm=False): argument152 def fetch_model(model_path, device, sparse_dlrm=False): argument
54 def measure_forward_pass(sparse_model_metadata, device, sparse_dlrm, **batch): argument