map tf.add_n to sum

Author: guschmue

tests/test_backend.py

@@ -770,6 +770,15 @@ def test_topk(self):
         actual, expected = self._run(output, {x: x_val}, {_INPUT: x_val})
         self.assertAllClose(expected, actual)
 
+    @unittest.skipIf(OPSET < 6, "supported since opset 6")
+    def test_addn(self):
+        x_val = np.arange(3*2*3).astype("float32")
+        x = tf.placeholder(tf.float32, x_val.shape, name=_TFINPUT)
+        x_ = tf.add_n([x, x, x])
+        output  = tf.identity(x_, name=_TFOUTPUT)
+        actual, expected = self._run(output, {x: x_val}, {_INPUT: x_val})
+        self.assertAllClose(expected, actual)
+
     @unittest.skipIf(BACKEND in ["caffe2", "onnxmsrt"], "multiple dims not supported")
     def test_strided_slice1(self):
         x_val = np.arange(3*2*3).astype("float32").reshape(3, 2, 3)

tf2onnx/tfonnx.py

@@ -876,6 +876,10 @@ def topk_op(ctx, node, name, args):
     "ExpandDims": (expanddims_op7, []),
 }
 
+_OPSET_6 = {
+    "AddN": (direct_op, ["Sum"]),
+}
+
 _OPSET_7 = {
     "Tile": (direct_op, []),
     "ResizeNearestNeighbor": (upsample_op, []),
@@ -903,6 +907,7 @@ def topk_op(ctx, node, name, args):
 _OPSETS = [
     (4, _OPSET_4),
     (5, _OPSET_5),
+    (6, _OPSET_6),
     (7, _OPSET_7),
 ]