diff --git a/README.md b/README.md index 14f29563..d2ad85e0 100644 --- a/README.md +++ b/README.md @@ -49,7 +49,7 @@ Self-Created Tools to convert ONNX files (NCHW) to TensorFlow format (NHWC). The $ docker run --rm -it \ -v `pwd`:/workdir \ -w /workdir \ -ghcr.io/pinto0309/onnx2tf:0.0.33 +ghcr.io/pinto0309/onnx2tf:1.0.0 or diff --git a/onnx2tf/__init__.py b/onnx2tf/__init__.py index 7fa6bd45..340aa4c6 100644 --- a/onnx2tf/__init__.py +++ b/onnx2tf/__init__.py @@ -1,3 +1,3 @@ from onnx2tf.onnx2tf import convert, main -__version__ = '0.0.33' +__version__ = '1.0.0' diff --git a/onnx2tf/ops/PRelu.py b/onnx2tf/ops/PRelu.py index d567468a..b659598c 100644 --- a/onnx2tf/ops/PRelu.py +++ b/onnx2tf/ops/PRelu.py @@ -10,6 +10,7 @@ inverted_operation_enable_disable, explicit_broadcast, make_tf_node_info, + channel_transpose, ) @@ -65,6 +66,11 @@ def make_node( y=slope, ) + slope = channel_transpose( + const_or_var_1=input_tensor, + const_or_var_2=slope, + ) + graph_node_output: gs.Variable = graph_node.outputs[0] shape = graph_node_output.shape dtype = graph_node_output.dtype