From 24d836b0f80141c4af0c557ae6ecc143ae4bdd9b Mon Sep 17 00:00:00 2001 From: Neo Chien Date: Sun, 25 Aug 2019 04:32:55 +0800 Subject: [PATCH] Fix code comment of operators (#3830) --- python/tvm/relay/frontend/onnx.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py index 11d73e26794c..07cda165cd12 100644 --- a/python/tvm/relay/frontend/onnx.py +++ b/python/tvm/relay/frontend/onnx.py @@ -222,6 +222,8 @@ def _impl_v1(cls, inputs, attr, params): class Div(Elemwise): + """ Operator converter for Divide. + """ name = 'divide' @@ -307,6 +309,8 @@ def _impl_v10(cls, inputs, attr, params): custom_check=dimension_constraint())(inputs, attr, params) class Mul(Elemwise): + """ Operator converter for Multiply. + """ name = 'multiply' @@ -495,6 +499,8 @@ def _impl_v1(cls, inputs, attr, params): class Sub(Elemwise): + """ Operator converter for Subtract. + """ name = 'subtract' @@ -768,27 +774,27 @@ def _impl_v1(cls, inputs, attr, params): return AttrCvt(cls.name)(inputs, attr) class ReduceMax(Reduce): - """ Operator converter for ArgMax. + """ Operator converter for ReduceMax. """ name = 'max' class ReduceMin(Reduce): - """ Operator converter for ArgMax. + """ Operator converter for ReduceMin. """ name = 'min' class ReduceSum(Reduce): - """ Operator converter for ArgMax. + """ Operator converter for ReduceSum. """ name = 'sum' class ReduceMean(Reduce): - """ Operator converter for ArgMax. + """ Operator converter for ReduceMean. """ name = 'mean' class ReduceProd(Reduce): - """ Operator converter for ArgMax. + """ Operator converter for ReduceProd. """ name = 'prod'