diff --git a/mace/python/tools/converter_tool/tensorflow_converter.py b/mace/python/tools/converter_tool/tensorflow_converter.py index da9384fa93affdca98ad05983055e35a6771b0e3..732abdcee5ef44ad5326556a5678ef2b21f9b852 100644 --- a/mace/python/tools/converter_tool/tensorflow_converter.py +++ b/mace/python/tools/converter_tool/tensorflow_converter.py @@ -54,6 +54,7 @@ TFSupportedOps = [ 'Mul', 'Div', 'Min', + 'Minimum', 'Max', 'Maximum', 'Neg', @@ -123,6 +124,7 @@ class TensorflowConverter(base_converter.ConverterInterface): TFOpType.Mul.name: EltwiseType.PROD, TFOpType.Div.name: EltwiseType.DIV, TFOpType.Min.name: EltwiseType.MIN, + TFOpType.Minimum.name: EltwiseType.MIN, TFOpType.Max.name: EltwiseType.MAX, TFOpType.Maximum.name: EltwiseType.MAX, TFOpType.Neg.name: EltwiseType.NEG, @@ -152,6 +154,7 @@ class TensorflowConverter(base_converter.ConverterInterface): TFOpType.Mul.name: self.convert_elementwise, TFOpType.Div.name: self.convert_elementwise, TFOpType.Min.name: self.convert_elementwise, + TFOpType.Minimum.name: self.convert_elementwise, TFOpType.Max.name: self.convert_elementwise, TFOpType.Maximum.name: self.convert_elementwise, TFOpType.Neg.name: self.convert_elementwise,