Searched refs:IntListWrapperModule (Results 1 – 1 of 1) sorted by relevance
15 class IntListWrapperModule(torch.nn.Module): class140 self.do_test_optional_intlist_with_module(IntListWrapperModule())141 self.do_test_optional_intlist_with_module(torch.jit.script(IntListWrapperModule()))159 IntListWrapperModule()(torch.zeros(1), [0.5])162 torch.jit.script(IntListWrapperModule())(torch.zeros(1), [0.5])165 IntListWrapperModule()(torch.zeros(1), torch.zeros(1))168 torch.jit.script(IntListWrapperModule())(torch.zeros(1), torch.zeros(1))