Home
last modified time | relevance | path

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

/external/pytorch/torch/ao/pruning/_experimental/data_sparsifier/benchmarks/
Ddlrm_utils.py25 def forward(self, dense_x, lS_o, lS_i): argument
106 def dlrm_wrap(X, lS_o, lS_i, device, ndevices=1): argument