Searched refs:SaveForwardInputsModule (Results 1 – 1 of 1) sorted by relevance
1197 class SaveForwardInputsModule(nn.Module): class1220 self.c1 = SaveForwardInputsModule(forward_inputs, cast_forward_inputs)1221 self.c2 = SaveForwardInputsModule(forward_inputs, cast_forward_inputs)