Home
last modified time | relevance | path

Searched refs:test_sparse_addmm (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_sparse_csr.py1959 def test_sparse_addmm(self, device, dtype): member in TestSparseCSR
H A Dtest_jit.py2105 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 Dtest_sparse.py1557 def test_sparse_addmm(self, device, dtype, coalesced): member in TestSparse