diff --git a/docs/api/python/contrib.rst b/docs/api/python/contrib.rst index a58a3aa4fbef..bc566759da99 100644 --- a/docs/api/python/contrib.rst +++ b/docs/api/python/contrib.rst @@ -77,6 +77,11 @@ tvm.contrib.rocm .. automodule:: tvm.contrib.rocm :members: +tvm.contrib.sparse +~~~~~~~~~~~~~~~~~~ +.. automodule:: tvm.contrib.sparse + :members: + tvm.contrib.spirv ~~~~~~~~~~~~~~~~~ diff --git a/docs/api/python/relay/frontend.rst b/docs/api/python/relay/frontend.rst index a418e042bf3d..054d3cecc1c5 100644 --- a/docs/api/python/relay/frontend.rst +++ b/docs/api/python/relay/frontend.rst @@ -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 diff --git a/docs/api/python/topi.rst b/docs/api/python/topi.rst index 450573e4c524..856bad198e88 100644 --- a/docs/api/python/topi.rst +++ b/docs/api/python/topi.rst @@ -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 ~~~~~~~~~~~~