[Doc]refine the example description of max/min/sum/tag_scope (#4974)

apache · Mar 2, 2020 · 892dc91 · 892dc91
1 parent 1c8e5b9
commit 892dc91
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/python/tvm/te/tag.py b/python/tvm/te/tag.py
@@ -86,6 +86,6 @@ def tag_scope(tag):
         # or use tag_scope as decorator
         @tvm.te.tag_scope(tag="conv")
         def compute_relu(data):
-            return te.compute(data.shape, lambda *i: tvm.select(data(*i) < 0, 0.0, data(*i)))
+            return te.compute(data.shape, lambda *i: tvm.tir.Select(data(*i) < 0, 0.0, data(*i)))
     """
     return TagScope(tag)
diff --git a/python/tvm/tir/op.py b/python/tvm/tir/op.py
@@ -950,6 +950,7 @@ def reducer(expr, axis, where=None, *args):
 
                 # there are two way to use this {0} reducer:
                 # mode 1, accept (expr, axis, where) to produce an Reduce Expr
+                # tvm.{0} represents tvm.te.{0} or tvm.tir.{0}.
                 B = te.compute((m,), lambda i: tvm.{0}(A[i, k], axis=k), name="B")
 
                 # mode 2, simply use it with multiple Exprs: