Searched refs:_CodeOnlyModule (Results 1 – 1 of 1) sorted by relevance
154 class _CodeOnlyModule(torch.nn.Module): class169 _CodeOnlyModule.forward = forward189 com = _CodeOnlyModule(body)431 if isinstance(root, _CodeOnlyModule):840 fake_mod = _CodeOnlyModule(copy.deepcopy(self.__dict__, memo))