Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_sparse.py4989 for sample in op.sample_inputs_sparse(layout, device, dtype):
5009 for sample in op.sample_inputs_sparse(layout, device, dtype, requires_grad=True):
5125 for sample in op.sample_inputs_sparse(layout, device, dtype):
5178 for sample in op.sample_inputs_sparse(layout, device, dtype):
/aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/
H A Dcore.py1316 def sample_inputs_sparse( member in OpInfo