added a type check for tf.float16

Ziyuan Lin · Ziyuan Lin · commit 617422770640 · 2018-08-09T17:41:58.000+03:00
diff --git a/tf2onnx/utils.py b/tf2onnx/utils.py
@@ -148,6 +148,8 @@ def get_tf_tensor_data(tensor):
         data = [0]
     elif tensor.dtype == tf.float32:
         data = [0.]
+    elif tensor.dtype == tf.float16:
+        data = [0]
     elif tensor.string_val:
         data = tensor.string_val
     else:
