Searched defs:select_dim (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse.py | 1160 def test_shape(sparse_dims, nnz, sizes, select_dim, select_index, fail_message=None): argument
1207 def test_shape(sparse_dims, nnz, sizes, select_dim, select_index, fail_message=None): argument
|