minor modification and remove test_sign_int

Author: mindest

tests/test_backend.py

@@ -1059,14 +1059,6 @@ def test_cast(self):
         _ = tf.identity(x_, name=_TFOUTPUT)
         self._run_test_case([_OUTPUT], {_INPUT: x_val})
 
-    @check_opset_min_version(9)
-    def test_sign_int(self):
-        x_val = np.array([1, 2, 0, -1, 0, -2], dtype=np.int).reshape((2, 3))
-        x = tf.placeholder(tf.int32, [2, 3], name=_TFINPUT)
-        x_ = tf.sign(x)
-        _ = tf.identity(x_, name=_TFOUTPUT)
-        self._run_test_case([_OUTPUT], {_INPUT: x_val})
-
     def test_sign(self):
         x_val1 = np.array([1, 2, 0, -1, 0, -2], dtype=np.int32).reshape((2, 3))
         x_val2 = np.array([1, 2, 0, -1, 0, -2], dtype=np.int64).reshape((2, 3))

tf2onnx/tfonnx.py

@@ -650,9 +650,9 @@ def sign_op4(ctx, node, name, args):
     if node_dtype in [onnx_pb.TensorProto.COMPLEX64, onnx_pb.TensorProto.COMPLEX128]:
         raise ValueError("dtype " + str(node_dtype) + " is not supported in onnx for now")
     zero_name = utils.make_name("{}_zero".format(node.name))
-    ctx.make_const(zero_name, np.array(0, dtype=utils.ONNX_TO_NUMPY_DTYPE[1]))
+    ctx.make_const(zero_name, np.array(0, dtype=np.float32))
     if node_dtype in [onnx_pb.TensorProto.INT32, onnx_pb.TensorProto.INT64]:
-        cast_node_0 = ctx.make_node("Cast", [node.input[0]], {"to": 1})
+        cast_node_0 = ctx.make_node("Cast", [node.input[0]], {"to": onnx_pb.TensorProto.FLOAT})
         greater_node = ctx.make_node("Greater", [cast_node_0.output[0], zero_name])
         less_node = ctx.make_node("Less", [cast_node_0.output[0], zero_name])
     else: