megengine.traced_module.register_as_builtin

register_as_builtin(mod_cls)

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

参数

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

返回类型

None

上一页

megengine.traced_module.trace_module

下一页

megengine.traced_module.wrap