Searched refs:StatefulWrapperModule (Results 1 – 1 of 1) sorted by relevance
232 class StatefulWrapperModule(torch.nn.Module): class272 StatefulWrapperModule(eager_model, getattr(eager_model, m_name)),