Searched refs:genSparseCSCTensor (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/pytorch/test/ |
H A D | test_sparse_csr.py | 1876 … 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 D | common_utils.py | 3348 def genSparseCSCTensor(self, size, nnz, *, device, dtype, index_dtype, dense_dims=0): member in TestCase
|