From 8cea9a6f96c03c1a8b6dd47ea82ebbeeacc95778 Mon Sep 17 00:00:00 2001 From: Lily Orth-Smith Date: Mon, 3 Aug 2020 18:51:25 -0700 Subject: [PATCH] [DOCS] Added casting to hybrid script doc and fixed pass infra doc (#6174) * updated hybridscript docs and pass infra docs * forgot uint16 --- docs/dev/hybrid_script.rst | 7 +++++++ docs/dev/relay_pass_infra.rst | 2 +- 2 files changed, 8 insertions(+), 1 deletion(-) 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() ])