This is an automated email from the ASF dual-hosted git repository.
ruihangl pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/main by this push:
new 726edbf54b [Docs] Complete API reference for tvm.relax backend and
testing modules (#19378)
726edbf54b is described below
commit 726edbf54bf816c041c7e23967b8fa0b9a86d8ab
Author: Shushi Hong <[email protected]>
AuthorDate: Fri Apr 10 00:30:26 2026 -0400
[Docs] Complete API reference for tvm.relax backend and testing modules
(#19378)
as per title
---
docs/reference/api/python/index.rst | 2 +
.../api/python/relax/{frontend.rst => backend.rst} | 61 ++++++++-----
docs/reference/api/python/relax/distributed.rst | 6 ++
docs/reference/api/python/relax/frontend.rst | 6 ++
.../python/relax/{distributed.rst => testing.rst} | 8 +-
docs/reference/api/python/relax/transform.rst | 101 +++++++++++++++++++++
6 files changed, 156 insertions(+), 28 deletions(-)
diff --git a/docs/reference/api/python/index.rst
b/docs/reference/api/python/index.rst
index 92c4f953a1..c826a87e89 100644
--- a/docs/reference/api/python/index.rst
+++ b/docs/reference/api/python/index.rst
@@ -44,10 +44,12 @@ Python API
relax/relax
relax/analysis
+ relax/backend
relax/block_builder
relax/distributed
relax/frontend
relax/op
+ relax/testing
relax/training
relax/transform
relax/dpl
diff --git a/docs/reference/api/python/relax/frontend.rst
b/docs/reference/api/python/relax/backend.rst
similarity index 54%
copy from docs/reference/api/python/relax/frontend.rst
copy to docs/reference/api/python/relax/backend.rst
index 7f3a2658b9..9c28499dd3 100644
--- a/docs/reference/api/python/relax/frontend.rst
+++ b/docs/reference/api/python/relax/backend.rst
@@ -15,40 +15,53 @@
specific language governing permissions and limitations
under the License.
-tvm.relax.frontend
-------------------
-.. automodule:: tvm.relax.frontend
+tvm.relax.backend
+-----------------
+
+tvm.relax.backend
+*****************
+.. automodule:: tvm.relax.backend
:members:
:imported-members:
-tvm.relax.frontend.nn
-*********************
-.. automodule:: tvm.relax.frontend.nn
+tvm.relax.backend.cuda
+**********************
+.. automodule:: tvm.relax.backend.cuda
:members:
- :imported-members:
- :exclude-members: BlockBuilder
- :noindex:
+ :no-index:
+
+tvm.relax.backend.rocm
+**********************
+.. automodule:: tvm.relax.backend.rocm
+ :members:
+ :no-index:
-tvm.relax.frontend.onnx
+tvm.relax.backend.metal
***********************
-.. automodule:: tvm.relax.frontend.onnx
+.. automodule:: tvm.relax.backend.metal
:members:
- :imported-members:
+ :no-index:
-tvm.relax.frontend.stablehlo
-****************************
-.. automodule:: tvm.relax.frontend.stablehlo
+tvm.relax.backend.adreno
+************************
+.. automodule:: tvm.relax.backend.adreno
:members:
- :imported-members:
+ :no-index:
-tvm.relax.frontend.tflite
-*************************
-.. automodule:: tvm.relax.frontend.tflite
+tvm.relax.backend.gpu_generic
+*****************************
+.. automodule:: tvm.relax.backend.gpu_generic
:members:
- :imported-members:
+ :no-index:
-tvm.relax.frontend.torch
-************************
-.. automodule:: tvm.relax.frontend.torch
+tvm.relax.backend.cpu_generic
+*****************************
+.. automodule:: tvm.relax.backend.cpu_generic
:members:
- :imported-members:
+ :no-index:
+
+tvm.relax.backend.contrib
+*************************
+.. automodule:: tvm.relax.backend.contrib
+ :members:
+ :no-index:
diff --git a/docs/reference/api/python/relax/distributed.rst
b/docs/reference/api/python/relax/distributed.rst
index a73012ccbf..82fc3a0445 100644
--- a/docs/reference/api/python/relax/distributed.rst
+++ b/docs/reference/api/python/relax/distributed.rst
@@ -20,3 +20,9 @@ tvm.relax.distributed
.. automodule:: tvm.relax.distributed
:members:
:imported-members:
+
+tvm.relax.distributed.transform
+*******************************
+.. automodule:: tvm.relax.distributed.transform
+ :members:
+ :no-index:
diff --git a/docs/reference/api/python/relax/frontend.rst
b/docs/reference/api/python/relax/frontend.rst
index 7f3a2658b9..1a83bda630 100644
--- a/docs/reference/api/python/relax/frontend.rst
+++ b/docs/reference/api/python/relax/frontend.rst
@@ -29,6 +29,12 @@ tvm.relax.frontend.nn
:exclude-members: BlockBuilder
:noindex:
+tvm.relax.frontend.nn.llm
+*************************
+.. automodule:: tvm.relax.frontend.nn.llm
+ :members:
+ :no-index:
+
tvm.relax.frontend.onnx
***********************
.. automodule:: tvm.relax.frontend.onnx
diff --git a/docs/reference/api/python/relax/distributed.rst
b/docs/reference/api/python/relax/testing.rst
similarity index 88%
copy from docs/reference/api/python/relax/distributed.rst
copy to docs/reference/api/python/relax/testing.rst
index a73012ccbf..aae52ee85f 100644
--- a/docs/reference/api/python/relax/distributed.rst
+++ b/docs/reference/api/python/relax/testing.rst
@@ -15,8 +15,8 @@
specific language governing permissions and limitations
under the License.
-tvm.relax.distributed
----------------------
-.. automodule:: tvm.relax.distributed
+tvm.relax.testing
+-----------------
+.. automodule:: tvm.relax.testing
:members:
- :imported-members:
+ :no-index:
diff --git a/docs/reference/api/python/relax/transform.rst
b/docs/reference/api/python/relax/transform.rst
index dcb41e80fd..f6c29ea4c3 100644
--- a/docs/reference/api/python/relax/transform.rst
+++ b/docs/reference/api/python/relax/transform.rst
@@ -22,3 +22,104 @@ tvm.relax.transform
.. automodule:: tvm.relax.transform
:members:
:imported-members:
+
+tvm.relax.transform.legalize_ops
+*********************************
+.. automodule:: tvm.relax.transform.legalize_ops
+ :members:
+
+tvm.relax.transform.legalize_ops.binary
+========================================
+.. automodule:: tvm.relax.transform.legalize_ops.binary
+ :members:
+ :no-index:
+
+tvm.relax.transform.legalize_ops.ccl
+=====================================
+.. automodule:: tvm.relax.transform.legalize_ops.ccl
+ :members:
+ :no-index:
+
+tvm.relax.transform.legalize_ops.create
+========================================
+.. automodule:: tvm.relax.transform.legalize_ops.create
+ :members:
+ :no-index:
+
+tvm.relax.transform.legalize_ops.datatype
+==========================================
+.. automodule:: tvm.relax.transform.legalize_ops.datatype
+ :members:
+ :no-index:
+
+tvm.relax.transform.legalize_ops.distributed
+==============================================
+.. automodule:: tvm.relax.transform.legalize_ops.distributed
+ :members:
+ :no-index:
+
+tvm.relax.transform.legalize_ops.grad
+======================================
+.. automodule:: tvm.relax.transform.legalize_ops.grad
+ :members:
+ :no-index:
+
+tvm.relax.transform.legalize_ops.image
+=======================================
+.. automodule:: tvm.relax.transform.legalize_ops.image
+ :members:
+ :no-index:
+
+tvm.relax.transform.legalize_ops.index
+=======================================
+.. automodule:: tvm.relax.transform.legalize_ops.index
+ :members:
+ :no-index:
+
+tvm.relax.transform.legalize_ops.linear_algebra
+================================================
+.. automodule:: tvm.relax.transform.legalize_ops.linear_algebra
+ :members:
+ :no-index:
+
+tvm.relax.transform.legalize_ops.manipulate
+=============================================
+.. automodule:: tvm.relax.transform.legalize_ops.manipulate
+ :members:
+ :no-index:
+
+tvm.relax.transform.legalize_ops.nn
+====================================
+.. automodule:: tvm.relax.transform.legalize_ops.nn
+ :members:
+ :no-index:
+
+tvm.relax.transform.legalize_ops.qdq
+=====================================
+.. automodule:: tvm.relax.transform.legalize_ops.qdq
+ :members:
+ :no-index:
+
+tvm.relax.transform.legalize_ops.search
+========================================
+.. automodule:: tvm.relax.transform.legalize_ops.search
+ :members:
+ :no-index:
+
+tvm.relax.transform.legalize_ops.statistical
+==============================================
+.. automodule:: tvm.relax.transform.legalize_ops.statistical
+ :members:
+ :no-index:
+
+tvm.relax.transform.legalize_ops.vision
+========================================
+.. automodule:: tvm.relax.transform.legalize_ops.vision
+ :members:
+ :no-index:
+
+tvm.relax.transform.legalize_ops.adreno
+========================================
+.. automodule:: tvm.relax.transform.legalize_ops.adreno
+ :members:
+ :no-index:
