fix comments

apache · Dec 22, 2020 · 3270634 · 3270634
1 parent a2277c3
commit 3270634
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/python/tvm/topi/nn/batch_matmul.py b/python/tvm/topi/nn/batch_matmul.py
@@ -64,12 +64,12 @@ def batch_matmul(x, y, oshape=None):
 
 @tvm.target.generic_func
 def batch_matmul_legalize(attrs, inputs, types):
-    """Legalizes Conv2D op.
+    """Legalizes batch_matmul op.
 
     Parameters
     ----------
     attrs : tvm.ir.Attrs
-        Attributes of current convolution
+        Attributes of current batch_matmul
     inputs : list of tvm.relay.Expr
         The args of the Relay expr to be legalized
     types : list of types

diff --git a/python/tvm/topi/nn/dense.py b/python/tvm/topi/nn/dense.py
@@ -67,12 +67,12 @@ def dense(data, weight, bias=None, out_dtype=None):
 
 @tvm.target.generic_func
 def dense_legalize(attrs, inputs, types):
-    """Legalizes Conv2D op.
+    """Legalizes dense op.
 
     Parameters
     ----------
     attrs : tvm.ir.Attrs
-        Attributes of current convolution
+        Attributes of current dense
     inputs : list of tvm.relay.Expr
         The args of the Relay expr to be legalized
     types : list of types