Skip to content

Commit 3c0973e

Browse files
committed
dtype.resource not supported on tf-1.x
1 parent 1ce0f99 commit 3c0973e

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

tf2onnx/tf_loader.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -201,12 +201,10 @@ def _from_saved_model_v1(sess, model_path, input_names, output_names, signatures
201201
for k in signatures:
202202
inputs_tensor_info = get_signature_def(imported, k).inputs
203203
for _, input_tensor in inputs_tensor_info.items():
204-
if input_tensor.dtype != tf.dtypes.resource:
205-
input_names.append(input_tensor.name)
204+
input_names.append(input_tensor.name)
206205
outputs_tensor_info = get_signature_def(imported, k).outputs
207206
for _, output_tensor in outputs_tensor_info.items():
208-
if output_tensor.dtype != tf.dtypes.resource:
209-
output_names.append(output_tensor.name)
207+
output_names.append(output_tensor.name)
210208
frozen_graph = freeze_session(sess, input_names=input_names, output_names=output_names)
211209
return frozen_graph, input_names, output_names
212210

0 commit comments

Comments
 (0)