Home
last modified time | relevance | path

Searched refs:genSparseTensor (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_cuda_multigpu.py1398 self.genSparseTensor((2, 3), 2, 1, False, "cuda", torch.float64)[0],
1401 self.genSparseTensor((2, 3), 2, 10, False, "cuda", torch.float64)[0],
1402 self.genSparseTensor((2, 3), 2, 5, False, "cuda", torch.float64)[0],
1403 self.genSparseTensor((3, 3), 2, 7, False, "cuda", torch.int64)[0],
1404 self.genSparseTensor((2, 3), 2, 2, False, "cuda", torch.float32)[0],
1407 self.genSparseTensor((2, 7), 2, 3, False, "cuda", torch.int64)[0],
1473 self.genSparseTensor((2, 3), 2, 1, False, "cuda", torch.float64)[0],
1476 self.genSparseTensor((2, 3), 2, 10, False, "cuda", torch.float64)[0],
1477 self.genSparseTensor((2, 3), 2, 5, False, "cuda", torch.float64)[0],
1478 self.genSparseTensor((3, 3), 2, 7, False, "cuda", torch.int64)[0],
[all …]
H A Dtest_sparse_csr.py1399 sparse_coo, _, _ = self.genSparseTensor(size, sparse_dim, nnz, coalesced, device, dtype)
3471 sparse, _, _ = self.genSparseTensor(shape, sparse_dim, nnz, coalesced, device, dtype)
H A Dtest_sparse.py196 …x, i, v = self.genSparseTensor(with_size, sparse_dim, nnz, not coalesced, dtype=dtype, device=devi…
H A Dtest_autograd.py11190 other = self.genSparseTensor(
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_utils.py3362 def genSparseTensor(self, size, sparse_dim, nnz, is_uncoalesced, device, dtype): member in TestCase