From 82ac680d3c73d67b916b4862f3072984b07b42cf Mon Sep 17 00:00:00 2001 From: liutuo Date: Fri, 24 Aug 2018 13:17:11 +0800 Subject: [PATCH] add support for tf.Minimum --- mace/python/tools/converter_tool/tensorflow_converter.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/mace/python/tools/converter_tool/tensorflow_converter.py b/mace/python/tools/converter_tool/tensorflow_converter.py index da9384fa..732abdce 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, -- GitLab