Commit 801068f3 by Yong Wu Committed by Tianqi Chen

update relay python api doc (#2766)

parent a803099f
......@@ -3,9 +3,6 @@ tvm.relay.backend
.. automodule:: tvm.relay.backend
Interpreter
-----------
.. automodule:: tvm.relay.backend.interpreter
:members:
......
......@@ -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:
......@@ -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
......
......@@ -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:
......@@ -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
......@@ -5,5 +5,5 @@ tvm.relay.image
.. automodule:: tvm.relay.image
:members:
.. automodule:: tvm.relay.op.image.image
.. automodule:: tvm.relay.op.image
:members:
......@@ -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
#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 *
......
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