Searched defs:dense_input_shape (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse_semi_structured.py | 183 def _test_mlp_contiguous_relu_compile(backend, dense_input_shape): argument
286 def test_mm_sparse_first_NN(self, dense_input_shape, dtype, device, backend): argument
313 def test_mm_sparse_first_NT(self, dense_input_shape, dtype, device, backend): argument
349 def test_mm_sparse_first_TN(self, dtype, dense_input_shape, device, backend): argument
370 def test_mm_sparse_second_NT(self, dense_input_shape, dtype, device, backend): argument
395 def test_mm_sparse_second_NN(self, dense_input_shape, dtype, device, backend): argument
416 def test_linear(self, dense_input_shape, inference_mode, device, backend): argument
444 def test_mlp(self, device, dense_input_shape, backend): argument
1042 def test_cslt_sparse_mm_mixed_dtype(self, dense_input_shape, out_dtype, device): argument
|