Searched refs:FunctionalModule (Results 1 – 3 of 3) sorted by relevance
312 class FunctionalModule(nn.Module): class338 return FunctionalModule(model_copy, param_names, names_map), params352 ) -> Tuple[FunctionalModule, Tuple[Tensor, ...]]:415 return FunctionalModule._create_from(
24 FunctionalModule,
167 class FunctionalModule(nn.Module): class170 return FunctionalModule