Home
last modified time | relevance | path

Searched refs:genSparseCSCTensor (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_sparse_csr.py1876 … x = self.genSparseCSCTensor((di, dk), nnz0, device=device, dtype=dtype, index_dtype=index_dtype)
1889 … y = self.genSparseCSCTensor((dk, dj), nnz1, device=device, dtype=dtype, index_dtype=index_dtype)
1894 gen_csr_csc = [self.genSparseCSRTensor, self.genSparseCSCTensor]
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_utils.py3348 def genSparseCSCTensor(self, size, nnz, *, device, dtype, index_dtype, dense_dims=0): member in TestCase