Searched refs:NNTransformerModule (Results 1 – 2 of 2) sorted by relevance
28 NNTransformerModule,80 "transformer_ops": NNTransformerModule(),
247 class NNTransformerModule(torch.nn.Module): class