modify for node_dtype

Author: mindest

tf2onnx/tfonnx.py

@@ -651,7 +651,7 @@ def sign_op4(ctx, node, name, args):
         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=np.float32))
-    if node_dtype in [onnx_pb.TensorProto.INT32, onnx_pb.TensorProto.INT64]:
+    if node_dtype not in [onnx_pb.TensorProto.FLOAT16, onnx_pb.TensorProto.FLOAT, onnx_pb.TensorProto.DOUBLE]:
         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])