From c6c9ee2f1f2b92c7cf7d667512f7d188e11c5261 Mon Sep 17 00:00:00 2001 From: Johan Verwey Date: Sat, 27 Mar 2021 10:56:06 -0700 Subject: [PATCH] Fix for export using TF2.3.1 --- keras2onnx/proto/tfcompat.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/keras2onnx/proto/tfcompat.py b/keras2onnx/proto/tfcompat.py index f82e5582..4d052f9e 100644 --- a/keras2onnx/proto/tfcompat.py +++ b/keras2onnx/proto/tfcompat.py @@ -10,9 +10,9 @@ def normalize_tensor_shape(tensor_shape): if is_tf2: - return [d for d in tensor_shape] - else: return [d.value for d in tensor_shape] + else: + return [d for d in tensor_shape] def dump_graph_into_tensorboard(tf_graph):