Searched refs:ModuleIfElse (Results 1 – 2 of 2) sorted by relevance
172 class ModuleIfElse(torch.nn.Module): class397 "module_if_else": lambda: (ModuleIfElse(), ModuleIfElse().get_random_inputs()),
224 class ModuleIfElse(nn.Module): class