megengine.traced_module.register_as_builtin¶

register_as_builtin(mod_cls)[源代码]¶

Registers class mod_cls (subclass of Module) as builtin module.

参数: mod_cls (Type[Module]) – the module class which will be treated as builtin module in tracing.
返回类型: None

megengine.traced_module.trace_module

megengine.traced_module.wrap