Commit 985e7d72 by Liangfu Chen Committed by Tianqi Chen

Update docs for some new modules (#2454)

parent d274e4b3
......@@ -77,6 +77,11 @@ tvm.contrib.rocm
.. automodule:: tvm.contrib.rocm
:members:
tvm.contrib.sparse
~~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.sparse
:members:
tvm.contrib.spirv
~~~~~~~~~~~~~~~~~
......
......@@ -5,3 +5,7 @@ tvm.relay.frontend
.. automodule:: tvm.relay.frontend
.. autofunction:: tvm.relay.frontend.from_mxnet
.. autofunction:: tvm.relay.frontend.from_keras
.. autofunction:: tvm.relay.frontend.from_onnx
......@@ -144,6 +144,11 @@ topi.image
~~~~~~~~~~
.. autofunction:: topi.image.resize
topi.sparse
~~~~~~~~~~~
.. autofunction:: topi.sparse.csrmv
.. autofunction:: topi.sparse.csrmm
.. autofunction:: topi.sparse.dense
topi.generic
~~~~~~~~~~~~
......
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