Searched refs:MaterializingGradFunction (Results 1 – 1 of 1) sorted by relevance
124 class MaterializingGradFunction(torch.autograd.Function): class137 return MaterializingGradFunction.apply(x)