Searched refs:genSparseCSRTensor (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse_csr.py | 1040 …a = self.genSparseCSRTensor((3, 3), 3, dtype=torch.float, device=self.device_type, index_dtype=tor…
1049 …a = self.genSparseCSRTensor((3, 3), 3, dtype=torch.float, device=self.device_type, index_dtype=tor…
1055 …a = self.genSparseCSRTensor((3, 3), 3, dtype=torch.float, device=self.device_type, index_dtype=tor…
1073 …a = self.genSparseCSRTensor((3, 3), 3, dtype=torch.float, device=self.device_type, index_dtype=tor…
1154 … a = self.genSparseCSRTensor(shape, nnz, dtype=dtype, device=device, index_dtype=index_dtype)
1305 … a = self.genSparseCSRTensor(shape, nnz, dtype=dtype, device=device, index_dtype=index_dtype)
1439 t = self.genSparseCSRTensor((m * blocksize, k * blocksize), nnz, dtype=dtype,
1456 t = self.genSparseCSRTensor((16, 16), nnz, dtype=dtype,
1491 …csr = self.genSparseCSRTensor((side, side), 1000, device=device, dtype=dtype, index_dtype=index_dt…
1535 … a = self.genSparseCSRTensor((m, k), nnz, dtype=dtype, device=device, index_dtype=index_dtype)
[all …]
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_utils.py | 3344 def genSparseCSRTensor(self, size, nnz, *, device, dtype, index_dtype, dense_dims=0): member in TestCase
|