Commit 70f0a169 by Neo Chien Committed by hlu1

Fix code comment of operators (#3830)

parent d05893b3
...@@ -222,6 +222,8 @@ class ConvTranspose(OnnxOpConverter): ...@@ -222,6 +222,8 @@ class ConvTranspose(OnnxOpConverter):
class Div(Elemwise): class Div(Elemwise):
""" Operator converter for Divide.
"""
name = 'divide' name = 'divide'
...@@ -307,6 +309,8 @@ class MaxPool(Pool): ...@@ -307,6 +309,8 @@ class MaxPool(Pool):
custom_check=dimension_constraint())(inputs, attr, params) custom_check=dimension_constraint())(inputs, attr, params)
class Mul(Elemwise): class Mul(Elemwise):
""" Operator converter for Multiply.
"""
name = 'multiply' name = 'multiply'
...@@ -495,6 +499,8 @@ class Softsign(OnnxOpConverter): ...@@ -495,6 +499,8 @@ class Softsign(OnnxOpConverter):
class Sub(Elemwise): class Sub(Elemwise):
""" Operator converter for Subtract.
"""
name = 'subtract' name = 'subtract'
...@@ -768,27 +774,27 @@ class Reduce(OnnxOpConverter): ...@@ -768,27 +774,27 @@ class Reduce(OnnxOpConverter):
return AttrCvt(cls.name)(inputs, attr) return AttrCvt(cls.name)(inputs, attr)
class ReduceMax(Reduce): class ReduceMax(Reduce):
""" Operator converter for ArgMax. """ Operator converter for ReduceMax.
""" """
name = 'max' name = 'max'
class ReduceMin(Reduce): class ReduceMin(Reduce):
""" Operator converter for ArgMax. """ Operator converter for ReduceMin.
""" """
name = 'min' name = 'min'
class ReduceSum(Reduce): class ReduceSum(Reduce):
""" Operator converter for ArgMax. """ Operator converter for ReduceSum.
""" """
name = 'sum' name = 'sum'
class ReduceMean(Reduce): class ReduceMean(Reduce):
""" Operator converter for ArgMax. """ Operator converter for ReduceMean.
""" """
name = 'mean' name = 'mean'
class ReduceProd(Reduce): class ReduceProd(Reduce):
""" Operator converter for ArgMax. """ Operator converter for ReduceProd.
""" """
name = 'prod' name = 'prod'
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment