Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_sparse.py13 class SparseTensorModule(torch.nn.Module): class
24 m = SparseTensorModule()
44 class SparseTensorModule(torch.nn.Module): class
54 m = SparseTensorModule()
68 class SparseTensorModule(torch.nn.Module): class
79 m = SparseTensorModule()
100 class SparseTensorModule(torch.nn.Module): class
110 m = SparseTensorModule()