diff --git a/docs/dev/hybrid_script.rst b/docs/dev/hybrid_script.rst index 7bb5e234cba0..939cf05bac6e 100644 --- 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 446a91bceff7..15487ac0227c 100644 --- 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() ])