Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_bundled_inputs.py29 class SingleTensorModel(torch.nn.Module): class
33 sm = torch.jit.script(SingleTensorModel())
85 class SingleTensorModel(torch.nn.Module): class
89 sm = torch.jit.script(SingleTensorModel())
105 class SingleTensorModel(torch.nn.Module): class
109 sm = torch.jit.script(SingleTensorModel())
261 class SingleTensorModel(torch.nn.Module): class
267 m = torch.jit.script(SingleTensorModel())
275 m = torch.jit.script(SingleTensorModel())
282 class SingleTensorModel(torch.nn.Module): class
[all …]
H A Dtest_bundled_images.py65 class SingleTensorModel(torch.nn.Module): class
73 sm = torch.jit.script(SingleTensorModel())