Merge pull request #466 from ziyuang/ziyuang-half-val-content-patch

consider half_val when retrieving tensor content
onnx · Apr 19, 2019 · d03e469 · d03e469
2 parents 5df0818 + 874def6
commit d03e469
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tf2onnx/utils.py b/tf2onnx/utils.py
@@ -165,6 +165,8 @@ def get_tf_tensor_data(tensor):
         is_raw = True
     elif tensor.float_val:
         data = tensor.float_val
+    elif tensor.half_val:
+        data = tensor.half_val
     elif tensor.dcomplex_val:
         data = tensor.dcomplex_val
     elif tensor.int_val: