Commit 801068f3 by Yong Wu Committed by Tianqi Chen

update relay python api doc (#2766)

parent a803099f
...@@ -3,9 +3,6 @@ tvm.relay.backend ...@@ -3,9 +3,6 @@ tvm.relay.backend
.. automodule:: tvm.relay.backend .. automodule:: tvm.relay.backend
Interpreter
-----------
.. automodule:: tvm.relay.backend.interpreter .. automodule:: tvm.relay.backend.interpreter
:members: :members:
......
...@@ -12,5 +12,8 @@ tvm.relay.base ...@@ -12,5 +12,8 @@ tvm.relay.base
.. autoclass:: tvm.relay.base.Span .. autoclass:: tvm.relay.base.Span
:members: :members:
.. autoclass:: tvm.relay.base.SourceName
:members:
.. autoclass:: tvm.relay.base.Id .. autoclass:: tvm.relay.base.Id
:members: :members:
...@@ -5,6 +5,8 @@ tvm.relay.build_module ...@@ -5,6 +5,8 @@ tvm.relay.build_module
.. autofunction:: tvm.relay.build_module.build .. autofunction:: tvm.relay.build_module.build
.. autofunction:: tvm.relay.build_module.build_config
.. autofunction:: tvm.relay.build_module.optimize .. autofunction:: tvm.relay.build_module.optimize
.. autofunction:: tvm.relay.build_module.create_executor .. autofunction:: tvm.relay.build_module.create_executor
......
...@@ -39,15 +39,21 @@ tvm.relay.expr ...@@ -39,15 +39,21 @@ tvm.relay.expr
.. autoclass:: tvm.relay.expr.TupleGetItem .. autoclass:: tvm.relay.expr.TupleGetItem
:members: :members:
.. autoclass:: tvm.relay.expr.TempExpr .. autoclass:: tvm.relay.expr.RefCreate
:members:
.. autoclass:: tvm.relay.expr.RefRead
:members:
.. autoclass:: tvm.relay.expr.RefWrite
:members: :members:
.. autoclass:: tvm.relay.expr.ExprFunctor .. autoclass:: tvm.relay.expr.TupleGetItem
:members: :members:
.. autoclass:: tvm.relay.expr.ExprMutator .. autoclass:: tvm.relay.expr.TempExpr
:members: :members:
.. autoclass:: tvm.relay.expr.TupleWrapper .. autoclass:: tvm.relay.expr.TupleWrapper
:members :members:
...@@ -9,3 +9,11 @@ tvm.relay.frontend ...@@ -9,3 +9,11 @@ tvm.relay.frontend
.. autofunction:: tvm.relay.frontend.from_keras .. autofunction:: tvm.relay.frontend.from_keras
.. autofunction:: tvm.relay.frontend.from_onnx .. 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
...@@ -5,5 +5,5 @@ tvm.relay.image ...@@ -5,5 +5,5 @@ tvm.relay.image
.. automodule:: tvm.relay.image .. automodule:: tvm.relay.image
:members: :members:
.. automodule:: tvm.relay.op.image.image .. automodule:: tvm.relay.op.image
:members: :members:
...@@ -3,8 +3,27 @@ tvm.relay.op ...@@ -3,8 +3,27 @@ tvm.relay.op
.. automodule:: tvm.relay.op .. automodule:: tvm.relay.op
:members: :members:
.. automodule:: tvm.relay.op.op .. autofunction:: tvm.relay.op.Op
:members:
.. 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 .. automodule:: tvm.relay.op.reduce
:members: :members:
...@@ -15,11 +34,10 @@ tvm.relay.op ...@@ -15,11 +34,10 @@ tvm.relay.op
.. automodule:: tvm.relay.op.transform .. automodule:: tvm.relay.op.transform
:members: :members:
.. automodule:: tvm.relay.op.nn.nn .. automodule:: tvm.relay.op.nn
:members: :members:
.. automodule:: tvm.relay.op.vision.multibox .. automodule:: tvm.relay.op.vision.multibox
:members: :members:
.. automodule:: tvm.relay.op.vision.nms .. autofunction:: tvm.relay.vision.nms
:members:
#pylint: disable=wildcard-import, redefined-builtin #pylint: disable=wildcard-import, redefined-builtin
"""Relay core operators.""" """Relay core operators."""
# operator defs # operator defs
from .op import get, register, register_schedule, register_compute, register_alter_op_layout, \ from .op import get, register, register_schedule, register_compute, register_gradient, \
Op register_pattern, register_alter_op_layout, schedule_injective, Op, OpPattern, debug
from .op import debug
# Operators # Operators
from .reduce import * from .reduce import *
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment