projects
/
platform
/
upstream
/
tvm.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
16fefd8
)
[Relay] Fix typo in ChangeBatch (#3660)
author
雾雨魔理沙
<lolisa@marisa.moe>
Tue, 30 Jul 2019 04:58:08 +0000
(21:58 -0700)
committer
Thierry Moreau
<moreau@uw.edu>
Tue, 30 Jul 2019 04:58:08 +0000
(21:58 -0700)
python/tvm/relay/transform.py
patch
|
blob
|
history
diff --git
a/python/tvm/relay/transform.py
b/python/tvm/relay/transform.py
index
8383759
..
2e64d14
100644
(file)
--- a/
python/tvm/relay/transform.py
+++ b/
python/tvm/relay/transform.py
@@
-920,7
+920,7
@@
class ChangeBatch:
----------
data: Dict[relay.Var, int]
A dictionary of all the params to change.
- The keys are all params, and the values
is
which dimension hold the batch.
+ The keys are all params, and the values
are
which dimension hold the batch.
batch_size: int
The batch size to change to.