Skip to content

Commit 874def6

Browse files
authored
consider half_val when retrieving tensor content
1 parent 6174227 commit 874def6

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
@@ -136,6 +136,8 @@ def get_tf_tensor_data(tensor):
136136
is_raw = True
137137
elif tensor.float_val:
138138
data = tensor.float_val
139+
elif tensor.half_val:
140+
data = tensor.half_val
139141
elif tensor.dcomplex_val:
140142
data = tensor.dcomplex_val
141143
elif tensor.int_val:

0 commit comments

Comments
 (0)