Support for channel_transpose

PINTO0309 · Oct 15, 2022 · 6514758 · 6514758
1 parent a0a7a19
commit 6514758
Show file tree

Hide file tree

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