Searched refs:CustomFuncBwdPrintModule (Results 1 – 1 of 1) sorted by relevance
151 class CustomFuncBwdPrintModule(torch.nn.Module): class271 model = CustomFuncBwdPrintModule()