Searched refs:NNLinearModule (Results 1 – 2 of 2) sorted by relevance
20 NNLinearModule,81 "linear_ops": NNLinearModule(),
273 class NNLinearModule(torch.nn.Module): class