Sqrt bug fixes

PINTO0309 · Jun 1, 2022 · 0b7b726 · 0b7b726
1 parent d771048
commit 0b7b726
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git 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
@@ -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",