[DOCS] Added casting to hybrid script doc and fixed pass infra doc (a…

…pache#6174) * updated hybridscript docs and pass infra docs * forgot uint16
trevor-m · Sep 2, 2020 · 8cea9a6 · 8cea9a6
1 parent c518a94
commit 8cea9a6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git 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
@@ -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()
     ])