diff --git a/python/tvm/relay/frontend/tflite.py b/python/tvm/relay/frontend/tflite.py index 55638e88a012a..be99c3474e144 100644 --- a/python/tvm/relay/frontend/tflite.py +++ b/python/tvm/relay/frontend/tflite.py @@ -714,6 +714,7 @@ def convert_div(self, op): return self._convert_elemwise(_op.divide, op) def convert_pow(self, op): + """Convert TFLite POW""" # Check if the input tensor is quantized, call QNN op if self.is_quantized(op): raise tvm.error.OpNotImplemented( @@ -721,6 +722,7 @@ def convert_pow(self, op): return self._convert_elemwise(_op.power, op) def convert_squared_difference(self, op): + """Convert TFLite SQUARED DIFFERENCE""" # Check if the input tensor is quantized, call QNN op if self.is_quantized(op): raise tvm.error.OpNotImplemented( @@ -732,6 +734,7 @@ def convert_squared_difference(self, op): return out def convert_maximum(self, op): + """Convert TFLite MAXIMUM""" # Check if the input tensor is quantized, call QNN op if self.is_quantized(op): raise tvm.error.OpNotImplemented( @@ -739,6 +742,7 @@ def convert_maximum(self, op): return self._convert_elemwise(_op.maximum, op) def convert_minimum(self, op): + """Convert TFLite MINIMUM""" # Check if the input tensor is quantized, call QNN op if self.is_quantized(op): raise tvm.error.OpNotImplemented( @@ -746,6 +750,7 @@ def convert_minimum(self, op): return self._convert_elemwise(_op.minimum, op) def convert_greater(self, op): + """Convert TFLite GREATER""" # Check if the input tensor is quantized, call QNN op if self.is_quantized(op): raise tvm.error.OpNotImplemented( @@ -753,30 +758,35 @@ def convert_greater(self, op): return self._convert_elemwise(_op.greater, op) def convert_greater_equal(self, op): + """Convert TFLite GREATER_EQUAL""" if self.is_quantized(op): raise tvm.error.OpNotImplemented( 'TFlite quantized GREATER_EQUAL operator is not supported yet.') return self._convert_elemwise(_op.greater_equal, op) def convert_less(self, op): + """Convert TFLite LESS""" if self.is_quantized(op): raise tvm.error.OpNotImplemented( 'TFlite quantized LESS operator is not supported yet.') return self._convert_elemwise(_op.less, op) def convert_less_equal(self, op): + """Convert TFLite LESS_EQUAL""" if self.is_quantized(op): raise tvm.error.OpNotImplemented( 'TFlite quantized LESS_EQUAL operator is not supported yet.') return self._convert_elemwise(_op.less_equal, op) def convert_equal(self, op): + """Convert TFLite EQUAL""" if self.is_quantized(op): raise tvm.error.OpNotImplemented( 'TFlite quantized EQUAL operator is not supported yet.') return self._convert_elemwise(_op.equal, op) def convert_not_equal(self, op): + """Convert TFLite NOT_EQUAL""" if self.is_quantized(op): raise tvm.error.OpNotImplemented( 'TFlite quantized NOT_EQUAL operator is not supported yet.')