Merge pull request #435 from chinhuang007/add-tf-onnx-ops

add direct ops from onnx-tf

Author: guschmue

tests/test_backend.py

@@ -1902,6 +1902,26 @@ def test_isnan(self):
             self._run_test_case([_OUTPUT], {_INPUT: x_val})
             tf.reset_default_graph()
 
+    def test_ceil(self):
+        x_val = np.array([-1.5, 1.2], dtype=np.float32)
+        x = tf.placeholder(tf.float32, [2], name=_TFINPUT)
+        x_ = tf.ceil(x)
+        _ = tf.identity(x_, name=_TFOUTPUT)
+        self._run_test_case([_OUTPUT], {_INPUT: x_val})
+
+    def test_softplus(self):
+        x_val = np.array([-1, 0, 1], dtype=np.float32)
+        x = tf.placeholder(tf.float32, [3], name=_TFINPUT)
+        x_ = tf.math.softplus(x)
+        _ = tf.identity(x_, name=_TFOUTPUT)
+        self._run_test_case([_OUTPUT], {_INPUT: x_val})
+
+    def test_softsign(self):
+        x_val = np.array([-1, 0, 1], dtype=np.float32)
+        x = tf.placeholder(tf.float32, [3], name=_TFINPUT)
+        x_ = tf.math.softsign(x)
+        _ = tf.identity(x_, name=_TFOUTPUT)
+        self._run_test_case([_OUTPUT], {_INPUT: x_val})
 
 if __name__ == '__main__':
     unittest_main()

tf2onnx/tfonnx.py

@@ -1756,6 +1756,7 @@ def where_op(ctx, node, name, args):
     "BiasAdd": (biasadd_op, []),
     "BiasAddV1": (biasadd_op, []),
     "Cast": (cast_op, []),
+    "Ceil": (direct_op, []),
     "CheckNumerics": (identity_op, ["Identity"]),
     "Concat": (concat_op, ["Concat"]),
     "ConcatV2": (concatv2_op, ["Concat"]),
@@ -1828,6 +1829,8 @@ def where_op(ctx, node, name, args):
     "Square": (square_op, []),
     "SquaredDifference": (squareddifference_op, []),
     "Softmax": (softmax_op, ["Softmax"]),
+    "Softplus": (direct_op, []),
+    "Softsign": (direct_op, []),
     "StopGradient": (identity_op, ["Identity"]),
     "StridedSlice": (stridedslice_op, []),
     "Sub": (broadcast_op, []),