Skip to content

Commit

Permalink
Fix code comment of operators (apache#3830)
Browse files Browse the repository at this point in the history
  • Loading branch information
cchung100m authored and wweic committed Sep 16, 2019
1 parent 5d03631 commit 24d836b
Showing 1 changed file with 11 additions and 5 deletions.
16 changes: 11 additions & 5 deletions python/tvm/relay/frontend/onnx.py
Original file line number Diff line number Diff line change
Expand Up @@ -222,6 +222,8 @@ def _impl_v1(cls, inputs, attr, params):


class Div(Elemwise):
""" Operator converter for Divide.
"""
name = 'divide'


Expand Down Expand Up @@ -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'


Expand Down Expand Up @@ -495,6 +499,8 @@ def _impl_v1(cls, inputs, attr, params):


class Sub(Elemwise):
""" Operator converter for Subtract.
"""
name = 'subtract'


Expand Down Expand Up @@ -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'

Expand Down

0 comments on commit 24d836b

Please sign in to comment.