Searched refs:TestNnModule (Results 1 – 3 of 3) sorted by relevance
16 from package_a.test_nn_module import TestNnModule18 test_nn_module = TestNnModule()19 test_torchscript_module = torch.jit.script(TestNnModule())20 test_fx_module: torch.fx.GraphModule = symbolic_trace(TestNnModule())
6 from package.package_a.test_nn_module import TestNnModule # noqa: F401
6 class TestNnModule(torch.nn.Module): class