diff --git a/README.md b/README.md index cd54bcf0..85a7c53e 100644 --- a/README.md +++ b/README.md @@ -299,7 +299,7 @@ Video speed is adjusted approximately 50 times slower than actual speed. docker run --rm -it \ -v `pwd`:/workdir \ -w /workdir \ - ghcr.io/pinto0309/onnx2tf:1.25.13 + ghcr.io/pinto0309/onnx2tf:1.25.14 or @@ -307,7 +307,7 @@ Video speed is adjusted approximately 50 times slower than actual speed. docker run --rm -it \ -v `pwd`:/workdir \ -w /workdir \ - docker.io/pinto0309/onnx2tf:1.25.13 + docker.io/pinto0309/onnx2tf:1.25.14 or diff --git a/onnx2tf/__init__.py b/onnx2tf/__init__.py index b162b232..f6db3dee 100644 --- a/onnx2tf/__init__.py +++ b/onnx2tf/__init__.py @@ -1,3 +1,3 @@ from onnx2tf.onnx2tf import convert, main -__version__ = '1.25.13' +__version__ = '1.25.14' diff --git a/onnx2tf/ops/ArgMin.py b/onnx2tf/ops/ArgMin.py index b5f27fbc..700e0d20 100644 --- a/onnx2tf/ops/ArgMin.py +++ b/onnx2tf/ops/ArgMin.py @@ -49,11 +49,15 @@ def make_node( shape = graph_node_output.shape dtype = graph_node_output.dtype + # Generation of TF OP + input_tensor = tf_layers_dict[graph_node_input.name]['tf_node'] \ + if isinstance(graph_node_input, gs.Variable) else graph_node_input + axis = graph_node.attrs.get('axis', 0) # NCHW->NHWC, NCDHW->NDHWC axis = convert_axis( axis=axis, - tensor_rank=len(graph_node_input.shape), + tensor_rank=len(graph_node_input.shape) if graph_node_input.shape is not None else len(input_tensor.shape), before_op_output_shape_trans=before_op_output_shape_trans, ) @@ -73,10 +77,6 @@ def make_node( and 'nhwc' in tf_layers_dict[graph_node_input.name].keys() else False } - # Generation of TF OP - input_tensor = tf_layers_dict[graph_node_input.name]['tf_node'] \ - if isinstance(graph_node_input, gs.Variable) else graph_node_input - # Pre-process transpose input_tensor = pre_process_transpose( value_before_transpose=input_tensor,