Searched refs:TestPyAssign (Results 1 – 1 of 1) sorted by relevance
703 class TestPyAssign(nn.Module): class711 TestPyAssign.__annotations__ = {"proxy_mod": OneTwo}714 scripted_pyassign_mod = torch.jit.script(TestPyAssign())