提交 e9a4cffe 编写于 作者: M Megvii Engine Team

fix(docs): fix traced module docs warning

GitOrigin-RevId: a5b790c7b527c700a594109d6d0219cd8b359e68
上级 b22ceba7
......@@ -21,10 +21,10 @@ from .traced_module import (
_register_all_builtin_module()
set_cpp_apply_module_trace(cpp_apply_module_trace)
__all__ = {
__all__ = [
"register_as_builtin",
"trace_module",
"wrap",
"TracedModule",
"optimize",
}
]
......@@ -222,7 +222,7 @@ class TensorNode(Node):
def __init__(
self,
expr: "Expr",
expr,
name: str = None,
qualname: str = None,
shape: Tuple[int] = None,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册