From 985e7d72bfc0f4bf486ac87c4a67b2dbb36d23f4 Mon Sep 17 00:00:00 2001 From: Liangfu Chen Date: Fri, 18 Jan 2019 01:08:30 +0800 Subject: [PATCH] Update docs for some new modules (#2454) --- docs/api/python/contrib.rst | 5 +++++ docs/api/python/relay/frontend.rst | 4 ++++ docs/api/python/topi.rst | 5 +++++ 3 files changed, 14 insertions(+) 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 ~~~~~~~~~~~~