megengine.traced_module.register_as_builtin

register_as_builtin(mod_cls)

将类 mod_cls （ Module 的子类）注册为内置模块。

参数:

mod_cls (Type[Module]) – module类在trace的时候会被当做内置模块。

返回类型:

None