From 0b7b726c3ed6fa1dca9606036d45381394cfb7a7 Mon Sep 17 00:00:00 2001 From: pinto0309 Date: Wed, 1 Jun 2022 22:20:11 +0900 Subject: [PATCH] Sqrt bug fixes --- openvino2tensorflow/openvino2tensorflow.py | 3 +-- setup.py | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/openvino2tensorflow/openvino2tensorflow.py b/openvino2tensorflow/openvino2tensorflow.py index 2f4b414..c962c7a 100644 --- a/openvino2tensorflow/openvino2tensorflow.py +++ b/openvino2tensorflow/openvino2tensorflow.py @@ -5754,8 +5754,7 @@ def depth_to_space(x, block_size): ### Sqrt elif layer.attrib['type'] == 'Sqrt': inp = tf.math.sqrt( - tf_layers_dict[get_tf_edges_from(tf_edges, layer_id, 0)], - tf_layers_dict[get_tf_edges_from(tf_edges, layer_id, 1)] + tf_layers_dict[get_tf_edges_from(tf_edges, layer_id, 0)] ) if wr_config and layer_id in wr_config and format_version >= 2: if wr_config[layer_id]['replace_mode'] == 'insert_before': diff --git a/setup.py b/setup.py index 25110e9..51c1c5a 100644 --- a/setup.py +++ b/setup.py @@ -11,7 +11,7 @@ setup( name="openvino2tensorflow", scripts=scripts, - version="1.31.2", + version="1.31.3", description="This script converts the OpenVINO IR model to Tensorflow's saved_model, tflite, h5 and pb. in (NCHW) format", long_description=long_description, long_description_content_type="text/markdown",