Searched refs:ModuleWithGradFunc (Results 1 – 1 of 1) sorted by relevance
216 class ModuleWithGradFunc(torch.nn.Module): class400 mod = ModuleWithGradFunc(ContextSaveAndMark)410 mod = ModuleWithGradFunc(ContextMarkAndSave)