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

fix(docs): fix traced module docs warning

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