[DOCS] Added casting to hybrid script doc and fixed pass infra doc (#6174)

* updated hybridscript docs and pass infra docs

* forgot uint16

diff --git a/docs/dev/hybrid_script.rst b/docs/dev/hybrid_script.rst
index 7bb5e23..939cf05 100644 (file)
--- a/docs/dev/hybrid_script.rst
+++ b/docs/dev/hybrid_script.rst
@@ -91,3 +91,10 @@ Math Intrinsics
 So far, these math intrinsics, ``log``, ``exp``, ``sigmoid``, ``tanh``, ``power``, and ``popcount``, are supported.
 Math intrinsics will be imported by the decorator. Most of the intrinsics are borrowed by library implementation
 except ``popcount`` and ``sigmoid``. I implemented them manually.
+
+
+Casting
+~~~~~~~
+
+You can cast values by using the keywords ``uint8``, ``uint16`` ``uint32``, ``uint64``, ``int8``, ``int16``, ``int32``, ``int64``, 
+``float16``, ``float32``, ``float64``.

diff --git a/docs/dev/relay_pass_infra.rst b/docs/dev/relay_pass_infra.rst
index 446a91b..15487ac 100644 (file)
--- a/docs/dev/relay_pass_infra.rst
+++ b/docs/dev/relay_pass_infra.rst
@@ -612,7 +612,7 @@ sequential pass example could be like the following to enable IR dumping for
     seq = tvm.transform.Sequential([
         relay.transform.InferType(),
         relay.transform.FoldConstant(),
-        transform.PrintIR(),
+        ir.transform.PrintIR(),
         relay.transform.EliminateCommonSubexpr(),
         relay.transform.AlterOpLayout()
     ])