Searched refs:test_sparse_addmm (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse_csr.py | 1959 def test_sparse_addmm(self, device, dtype): member in TestSparseCSR
|
| H A D | test_jit.py | 2105 def test_sparse_addmm(input, input1, input2): function
2111 self.checkScript(test_sparse_addmm, (torch.randn(2, 4), get_sparse(), torch.randn(3, 4)))
|
| H A D | test_sparse.py | 1557 def test_sparse_addmm(self, device, dtype, coalesced): member in TestSparse
|