From ab7499a5ff6eb8283f92e3381857a22cad62190d Mon Sep 17 00:00:00 2001 From: Riccardo <36033044+gasgallo@users.noreply.github.com> Date: Fri, 8 May 2020 15:58:34 +0700 Subject: [PATCH] Add support for AddV2 op in TF (#637) Co-authored-by: riccardo --- tools/python/transform/tensorflow_converter.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tools/python/transform/tensorflow_converter.py b/tools/python/transform/tensorflow_converter.py index 73a62dd7..c57a83fc 100644 --- a/tools/python/transform/tensorflow_converter.py +++ b/tools/python/transform/tensorflow_converter.py @@ -54,6 +54,7 @@ tf_axis = 'axis' TFSupportedOps = [ 'Abs', 'Add', + 'AddV2', 'ArgMax', 'AvgPool', 'BatchMatMul', @@ -165,6 +166,7 @@ class TensorflowConverter(base_converter.ConverterInterface): eltwise_type = { TFOpType.Add.name: EltwiseType.SUM, + TFOpType.AddV2.name: EltwiseType.SUM, TFOpType.Sub.name: EltwiseType.SUB, TFOpType.Mul.name: EltwiseType.PROD, TFOpType.Div.name: EltwiseType.DIV, @@ -210,6 +212,7 @@ class TensorflowConverter(base_converter.ConverterInterface): self._op_converters = { TFOpType.Abs.name: self.convert_elementwise, TFOpType.Add.name: self.convert_add, + TFOpType.AddV2.name: self.convert_add, TFOpType.ArgMax.name: self.convert_argmax, TFOpType.AvgPool.name: self.convert_pooling, TFOpType.BatchMatMul.name: self.convert_matmul, -- GitLab