[DOCS] Fix tag_scope example (#581)

18e4a1bd · Wei Chen · Tianqi Chen · 12218358 · 18e4a1bd
Commit 18e4a1bd authored Oct 24, 2017 by Wei Chen Committed by Tianqi Chen Oct 24, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

python/tvm/tag.py
+2 -2

No files found.
--- a/python/tvm/tag.py
+++ b/python/tvm/tag.py
@@ -54,9 +54,9 @@ def tag_scope(tag):

        n = tvm.var('n')
        m = tvm.var('m')
-        l = tvm.var('m')
+        l = tvm.var('l')
        A = tvm.placeholder((n, l), name='A')
-        B = tvm.placeholder((m, l), name='A')
+        B = tvm.placeholder((m, l), name='B')
        k = tvm.reduce_axis((0, l), name='k')

        with tvm.tag_scope(tag='matmul'):