Searched refs:sample_inputs_sparse_like_fns (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/opinfo/definitions/ |
| H A D | sparse.py | 855 def sample_inputs_sparse_like_fns( function
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_methods_invocations.py | 142 sample_inputs_sparse_like_fns,
18608 … sample_inputs_sparse_coo_func=partial(sample_inputs_sparse_like_fns, layout=torch.sparse_coo),
18609 … sample_inputs_sparse_csr_func=partial(sample_inputs_sparse_like_fns, layout=torch.sparse_csr),
18610 … sample_inputs_sparse_csc_func=partial(sample_inputs_sparse_like_fns, layout=torch.sparse_csc),
18611 … sample_inputs_sparse_bsr_func=partial(sample_inputs_sparse_like_fns, layout=torch.sparse_bsr),
18612 … sample_inputs_sparse_bsc_func=partial(sample_inputs_sparse_like_fns, layout=torch.sparse_bsc),
18656 … sample_inputs_sparse_coo_func=partial(sample_inputs_sparse_like_fns, layout=torch.sparse_coo),
18657 … sample_inputs_sparse_csr_func=partial(sample_inputs_sparse_like_fns, layout=torch.sparse_csr),
18658 … sample_inputs_sparse_csc_func=partial(sample_inputs_sparse_like_fns, layout=torch.sparse_csc),
18659 … sample_inputs_sparse_bsr_func=partial(sample_inputs_sparse_like_fns, layout=torch.sparse_bsr),
[all …]
|