Home
last modified time | relevance | path

Searched refs:SparseSemiStructuredTensorCUSPARSELT (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_sparse_semi_structured.py12 SparseSemiStructuredTensorCUSPARSELT,
60 SEMI_STRUCTURED_SUPPORTED_BACKENDS["cusparselt"] = SparseSemiStructuredTensorCUSPARSELT
250 y = SparseSemiStructuredTensorCUSPARSELT.prune_dense_static_sort(x)
600 reference_cusparselt = SparseSemiStructuredTensorCUSPARSELT.prune_dense_static_sort(pruned,
606 cusparselt = SparseSemiStructuredTensorCUSPARSELT(dense.shape,
/aosp_15_r20/external/pytorch/torch/sparse/
H A Dsemi_structured.py378 else torch.sparse.SparseSemiStructuredTensorCUSPARSELT
523 class SparseSemiStructuredTensorCUSPARSELT(SparseSemiStructuredTensor): class
H A D__init__.py13 SparseSemiStructuredTensorCUSPARSELT,