Skip to content
Projects
Groups
Snippets
Help
This project
Loading...
Sign in / Register
Toggle navigation
T
tic
Overview
Overview
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
wenyuanbo
tic
Commits
892dc91a
Unverified
Commit
892dc91a
authored
Mar 02, 2020
by
Ethan-Yan27
Committed by
GitHub
Mar 01, 2020
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
[Doc]refine the example description of max/min/sum/tag_scope (#4974)
parent
1c8e5b93
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
1 deletions
+2
-1
python/tvm/te/tag.py
+1
-1
python/tvm/tir/op.py
+1
-0
No files found.
python/tvm/te/tag.py
View file @
892dc91a
...
...
@@ -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.
s
elect(data(*i) < 0, 0.0, data(*i)))
return te.compute(data.shape, lambda *i: tvm.
tir.S
elect(data(*i) < 0, 0.0, data(*i)))
"""
return
TagScope
(
tag
)
python/tvm/tir/op.py
View file @
892dc91a
...
...
@@ -950,6 +950,7 @@ def comm_reducer(fcombine, fidentity, name="reduce"):
# 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:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment