Skip to content

Commit d03e469

Browse files
authored
Merge pull request #466 from ziyuang/ziyuang-half-val-content-patch
consider half_val when retrieving tensor content
2 parents 5df0818 + 874def6 commit d03e469

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tf2onnx/utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -165,6 +165,8 @@ def get_tf_tensor_data(tensor):
165165
is_raw = True
166166
elif tensor.float_val:
167167
data = tensor.float_val
168+
elif tensor.half_val:
169+
data = tensor.half_val
168170
elif tensor.dcomplex_val:
169171
data = tensor.dcomplex_val
170172
elif tensor.int_val:

0 commit comments

Comments
 (0)