Searched refs:genSparseCompressedTensor (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse_csr.py | 204 …return self.genSparseCompressedTensor(size, nnz, device=device, dtype=dtype, index_dtype=index_dty…
469 … a = self.genSparseCompressedTensor(shape, nnz, dtype=dtype, layout=layout, device=device,
471 … b = self.genSparseCompressedTensor(shape, nnz, dtype=dtype, layout=layout, device=device,
493 a = self.genSparseCompressedTensor(shape1, 0, dtype=dtype, layout=layout, device=device,
500 … b = self.genSparseCompressedTensor(shape1, nnz, dtype=dtype, layout=layout, device=device,
508 … c = self.genSparseCompressedTensor(shape2, nnz, dtype=dtype, layout=layout, device=device,
517 … d = self.genSparseCompressedTensor(shape1, nnz, dtype=dtype, layout=layout, device=device,
1103 sparse = self.genSparseCompressedTensor(
1233 a = self.genSparseCompressedTensor(shape,
1242 b = self.genSparseCompressedTensor(shape,
[all …]
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_utils.py | 3298 …def genSparseCompressedTensor(self, size, nnz, *, layout, device, dtype, index_dtype, blocksize=()… member in TestCase
3345 return self.genSparseCompressedTensor(size, nnz, layout=torch.sparse_csr, device=device,
3349 return self.genSparseCompressedTensor(size, nnz, layout=torch.sparse_csc, device=device,
3354 return self.genSparseCompressedTensor(size, nnz, layout=torch.sparse_bsr, device=device,
3359 return self.genSparseCompressedTensor(size, nnz, layout=torch.sparse_bsc, device=device,
|