update relay python api doc (#2766)

apache · Mar 11, 2019 · 801068f · 801068f
1 parent a803099
commit 801068f
Show file tree

Hide file tree

Showing 8 changed files with 49 additions and 16 deletions.
diff --git a/docs/api/python/relay/backend.rst b/docs/api/python/relay/backend.rst
@@ -3,9 +3,6 @@ tvm.relay.backend
 
 .. automodule:: tvm.relay.backend
 
-Interpreter
------------
-
 .. automodule:: tvm.relay.backend.interpreter
     :members:
 

diff --git a/docs/api/python/relay/base.rst b/docs/api/python/relay/base.rst
@@ -12,5 +12,8 @@ tvm.relay.base
 .. autoclass:: tvm.relay.base.Span
     :members:
 
+.. autoclass:: tvm.relay.base.SourceName
+    :members:
+
 .. autoclass:: tvm.relay.base.Id
     :members:
diff --git a/docs/api/python/relay/build_module.rst b/docs/api/python/relay/build_module.rst
@@ -5,6 +5,8 @@ tvm.relay.build_module
 
 .. autofunction:: tvm.relay.build_module.build
 
+.. autofunction:: tvm.relay.build_module.build_config
+
 .. autofunction:: tvm.relay.build_module.optimize
 
 .. autofunction:: tvm.relay.build_module.create_executor

diff --git a/docs/api/python/relay/expr.rst b/docs/api/python/relay/expr.rst
@@ -39,15 +39,21 @@ tvm.relay.expr
 .. autoclass:: tvm.relay.expr.TupleGetItem
     :members:
 
-.. autoclass:: tvm.relay.expr.TempExpr
+.. autoclass:: tvm.relay.expr.RefCreate
+    :members:
+
+.. autoclass:: tvm.relay.expr.RefRead
+    :members:
+
+.. autoclass:: tvm.relay.expr.RefWrite
     :members:
 
-.. autoclass:: tvm.relay.expr.ExprFunctor
+.. autoclass:: tvm.relay.expr.TupleGetItem
     :members:
 
-.. autoclass:: tvm.relay.expr.ExprMutator
+.. autoclass:: tvm.relay.expr.TempExpr
     :members:
 
 .. autoclass:: tvm.relay.expr.TupleWrapper
-    :members
+    :members:
 
diff --git a/docs/api/python/relay/frontend.rst b/docs/api/python/relay/frontend.rst
@@ -9,3 +9,11 @@ tvm.relay.frontend
 .. autofunction:: tvm.relay.frontend.from_keras
 
 .. autofunction:: tvm.relay.frontend.from_onnx
+
+.. autofunction:: tvm.relay.frontend.from_tflite
+
+.. autofunction:: tvm.relay.frontend.from_coreml
+
+.. autofunction:: tvm.relay.frontend.from_caffe2
+
+.. autofunction:: tvm.relay.frontend.from_tensorflow
diff --git a/docs/api/python/relay/image.rst b/docs/api/python/relay/image.rst
@@ -5,5 +5,5 @@ tvm.relay.image
 .. automodule:: tvm.relay.image
     :members:
 
-.. automodule:: tvm.relay.op.image.image
+.. automodule:: tvm.relay.op.image
     :members:
diff --git a/docs/api/python/relay/op.rst b/docs/api/python/relay/op.rst
@@ -3,8 +3,27 @@ tvm.relay.op
 .. automodule:: tvm.relay.op
     :members:
 
-.. automodule:: tvm.relay.op.op
-    :members:
+.. autofunction:: tvm.relay.op.Op
+
+.. autofunction:: tvm.relay.op.OpPattern
+
+.. autofunction:: tvm.relay.op.get
+
+.. autofunction:: tvm.relay.op.register
+
+.. autofunction:: tvm.relay.op.register_schedule
+
+.. autofunction:: tvm.relay.op.register_pattern
+
+.. autofunction:: tvm.relay.op.register_compute
+
+.. autofunction:: tvm.relay.op.register_gradient
+
+.. autofunction:: tvm.relay.op.register_alter_op_layout
+
+.. autofunction:: tvm.relay.op.schedule_injective
+
+.. autofunction:: tvm.relay.op.debug
 
 .. automodule:: tvm.relay.op.reduce
     :members:
@@ -15,11 +34,10 @@ tvm.relay.op
 .. automodule:: tvm.relay.op.transform
     :members:
 
-.. automodule:: tvm.relay.op.nn.nn
+.. automodule:: tvm.relay.op.nn
     :members:
 
 .. automodule:: tvm.relay.op.vision.multibox
     :members:
 
-.. automodule:: tvm.relay.op.vision.nms
-    :members:
+.. autofunction:: tvm.relay.vision.nms
diff --git a/python/tvm/relay/op/__init__.py b/python/tvm/relay/op/__init__.py
@@ -1,9 +1,8 @@
 #pylint: disable=wildcard-import, redefined-builtin
 """Relay core operators."""
 # operator defs
-from .op import get, register, register_schedule, register_compute, register_alter_op_layout, \
-    Op
-from .op import debug
+from .op import get, register, register_schedule, register_compute, register_gradient, \
+    register_pattern, register_alter_op_layout, schedule_injective, Op, OpPattern, debug
 
 # Operators
 from .reduce import *