From 32706340511882896b8fdcf232cb12e8f56f43da Mon Sep 17 00:00:00 2001 From: "liuxin.ai" Date: Tue, 22 Dec 2020 14:42:19 +0800 Subject: [PATCH] fix comments --- python/tvm/topi/nn/batch_matmul.py | 4 ++-- python/tvm/topi/nn/dense.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/python/tvm/topi/nn/batch_matmul.py b/python/tvm/topi/nn/batch_matmul.py index 9fd795807dc4..3cd3c53c3bcb 100644 --- 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 index 93068336180f..348a6098d99f 100644 --- 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