Searched refs:wrap_module (Results 1 – 2 of 2) sorted by relevance
83 def wrap_module(module: nn.Module) -> nn.Module: function90 wrap_module(model.seq1)
651 return self.wrap_module(value)1317 def wrap_module(self, value: torch.nn.Module): member in VariableBuilder1333 return self.wrap_module(value._orig_mod)