Searched refs:ModuleDirectforwardSubmodCall (Results 1 – 2 of 2) sorted by relevance
27 ModuleDirectforwardSubmodCall,198 m_submod_forward_call = ModuleDirectforwardSubmodCall(
52 class ModuleDirectforwardSubmodCall(torch.nn.Module): class