Searched refs:ModuleIfWhile (Results 1 – 2 of 2) sorted by relevance
271 class ModuleIfWhile(torch.nn.Module): class403 "module_if_while": lambda: (ModuleIfWhile(), ModuleIfWhile().get_random_inputs()),
323 class ModuleIfWhile(nn.Module): class