onnx2tf 1.29.18__tar.gz → 1.29.20__tar.gz

{onnx2tf-1.29.18 → onnx2tf-1.29.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnx2tf
-Version: 1.29.18
+Version: 1.29.20
 Summary: Self-Created Tools to convert ONNX files (NCHW) to TensorFlow/TFLite/Keras format (NHWC). The purpose of this tool is to solve the massive Transpose extrapolation problem in onnx-tensorflow (onnx-tf).
 Keywords: onnx,tensorflow,tflite,keras,deep-learning,machine-learning
 Author: Katsuya Hyodo
@@ -137,11 +137,11 @@ https://github.com/PINTO0309/onnx2tf/wiki/model_status
   |Cos|:heavy_check_mark:|
   |CumProd|:heavy_check_mark:|
   |CumSum|:heavy_check_mark:|
-  |DeformConv|**Help wanted**|
+  |DeformConv|:white_check_mark:|
   |DepthToSpace|:heavy_check_mark:|
   |Det|:heavy_check_mark:|
   |DequantizeLinear|:heavy_check_mark:|
-  |DFT|**Help wanted**|
+  |DFT|:white_check_mark:|
   |Div|:heavy_check_mark:|
   |Dropout|:heavy_check_mark:|
   |DynamicQuantizeLinear|:heavy_check_mark:|
@@ -365,7 +365,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.29.18
+  ghcr.io/pinto0309/onnx2tf:1.29.20
 
   or
 
@@ -373,7 +373,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.29.18
+  docker.io/pinto0309/onnx2tf:1.29.20
 
   or
 
@@ -1887,6 +1887,15 @@ optional arguments:
     model partitioned into subgraphs.
     e.g. --output_names_to_interrupt_model_conversion "output0" "output1" "output2"
 
+  -easm, --enable_auto_split_model
+    Force auto split regardless of the ONNX file size.
+    Uses --auto_split_max_size_mb as the target partition size.
+
+  -asmsm AUTO_SPLIT_MAX_SIZE_MB, --auto_split_max_size_mb AUTO_SPLIT_MAX_SIZE_MB
+    Target maximum size per partition in MB based on ONNX initializer sizes.
+    Used when auto-split is triggered or forced.
+    Default: 1024
+
   -dgc, --disable_group_convolution
     Disable GroupConvolution and replace it with SeparableConvolution for
     output to saved_model format.
@@ -2156,6 +2165,8 @@ convert(
   keep_shape_absolutely_input_names: Optional[List[str]] = None,
   input_names_to_interrupt_model_conversion: Union[List[str], NoneType] = None,
   output_names_to_interrupt_model_conversion: Union[List[str], NoneType] = None,
+  enable_auto_split_model: Optional[bool] = False,
+  auto_split_max_size_mb: Optional[int] = 1024,
   disable_group_convolution: Union[bool, NoneType] = False,
   enable_batchmatmul_unfold: Optional[bool] = False,
   enable_rnn_unroll: Optional[bool] = False,
@@ -2424,6 +2435,17 @@ convert(
       e.g.
       output_names_to_interrupt_model_conversion=['output0','output1','output2']
 
+    enable_auto_split_model: Optional[bool]
+      Force auto split regardless of the ONNX file size.
+      Uses auto_split_max_size_mb as the target partition size.
+      Short option: -easm
+      Default: False
+
+    auto_split_max_size_mb: Optional[int]
+      Target maximum size per partition in MB based on ONNX initializer sizes.
+      Used when auto-split is triggered or forced.
+      Default: 1024
+
     disable_group_convolution: Optional[bool]
       Disable GroupConvolution and replace it with SeparableConvolution for
       output to saved_model format.

{onnx2tf-1.29.18 → onnx2tf-1.29.20}/README.md

@@ -95,11 +95,11 @@ https://github.com/PINTO0309/onnx2tf/wiki/model_status
   |Cos|:heavy_check_mark:|
   |CumProd|:heavy_check_mark:|
   |CumSum|:heavy_check_mark:|
-  |DeformConv|**Help wanted**|
+  |DeformConv|:white_check_mark:|
   |DepthToSpace|:heavy_check_mark:|
   |Det|:heavy_check_mark:|
   |DequantizeLinear|:heavy_check_mark:|
-  |DFT|**Help wanted**|
+  |DFT|:white_check_mark:|
   |Div|:heavy_check_mark:|
   |Dropout|:heavy_check_mark:|
   |DynamicQuantizeLinear|:heavy_check_mark:|
@@ -323,7 +323,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.29.18
+  ghcr.io/pinto0309/onnx2tf:1.29.20
 
   or
 
@@ -331,7 +331,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.29.18
+  docker.io/pinto0309/onnx2tf:1.29.20
 
   or
 
@@ -1845,6 +1845,15 @@ optional arguments:
     model partitioned into subgraphs.
     e.g. --output_names_to_interrupt_model_conversion "output0" "output1" "output2"
 
+  -easm, --enable_auto_split_model
+    Force auto split regardless of the ONNX file size.
+    Uses --auto_split_max_size_mb as the target partition size.
+
+  -asmsm AUTO_SPLIT_MAX_SIZE_MB, --auto_split_max_size_mb AUTO_SPLIT_MAX_SIZE_MB
+    Target maximum size per partition in MB based on ONNX initializer sizes.
+    Used when auto-split is triggered or forced.
+    Default: 1024
+
   -dgc, --disable_group_convolution
     Disable GroupConvolution and replace it with SeparableConvolution for
     output to saved_model format.
@@ -2114,6 +2123,8 @@ convert(
   keep_shape_absolutely_input_names: Optional[List[str]] = None,
   input_names_to_interrupt_model_conversion: Union[List[str], NoneType] = None,
   output_names_to_interrupt_model_conversion: Union[List[str], NoneType] = None,
+  enable_auto_split_model: Optional[bool] = False,
+  auto_split_max_size_mb: Optional[int] = 1024,
   disable_group_convolution: Union[bool, NoneType] = False,
   enable_batchmatmul_unfold: Optional[bool] = False,
   enable_rnn_unroll: Optional[bool] = False,
@@ -2382,6 +2393,17 @@ convert(
       e.g.
       output_names_to_interrupt_model_conversion=['output0','output1','output2']
 
+    enable_auto_split_model: Optional[bool]
+      Force auto split regardless of the ONNX file size.
+      Uses auto_split_max_size_mb as the target partition size.
+      Short option: -easm
+      Default: False
+
+    auto_split_max_size_mb: Optional[int]
+      Target maximum size per partition in MB based on ONNX initializer sizes.
+      Used when auto-split is triggered or forced.
+      Default: 1024
+
     disable_group_convolution: Optional[bool]
       Disable GroupConvolution and replace it with SeparableConvolution for
       output to saved_model format.

{onnx2tf-1.29.18 → onnx2tf-1.29.20}/onnx2tf/__init__.py

@@ -1,3 +1,3 @@
 from onnx2tf.onnx2tf import convert, main
 
-__version__ = '1.29.18'
+__version__ = '1.29.20'