Module-class composite opsΒΆ
nn.Module subclasses exposed in coreai_torch.composite_ops. Build these into your model as named submodules and externalize them with an ExternalizeSpec, passing them to the externalize_modules parameter of add_pytorch_module(). For a tutorial walkthrough, see Composite Ops Guide.