onnx2tf 1.25.13__py3-none-any.whl → 1.25.15__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- onnx2tf/__init__.py +1 -1
- onnx2tf/ops/Add.py +4 -0
- onnx2tf/ops/ArgMin.py +5 -5
- onnx2tf/ops/Mul.py +4 -0
- {onnx2tf-1.25.13.dist-info → onnx2tf-1.25.15.dist-info}/METADATA +3 -3
- {onnx2tf-1.25.13.dist-info → onnx2tf-1.25.15.dist-info}/RECORD +11 -11
- {onnx2tf-1.25.13.dist-info → onnx2tf-1.25.15.dist-info}/LICENSE +0 -0
- {onnx2tf-1.25.13.dist-info → onnx2tf-1.25.15.dist-info}/LICENSE_onnx-tensorflow +0 -0
- {onnx2tf-1.25.13.dist-info → onnx2tf-1.25.15.dist-info}/WHEEL +0 -0
- {onnx2tf-1.25.13.dist-info → onnx2tf-1.25.15.dist-info}/entry_points.txt +0 -0
- {onnx2tf-1.25.13.dist-info → onnx2tf-1.25.15.dist-info}/top_level.txt +0 -0
onnx2tf/__init__.py
CHANGED
onnx2tf/ops/Add.py
CHANGED
|
@@ -92,6 +92,10 @@ def make_node(
|
|
|
92
92
|
input_tensor_2 = tf_layers_dict[graph_node_input_2.name]['tf_node'] \
|
|
93
93
|
if isinstance(graph_node_input_2, gs.Variable) else graph_node_input_2
|
|
94
94
|
|
|
95
|
+
# issue: https://github.com/PINTO0309/onnx2tf/issues/698
|
|
96
|
+
if isinstance(input_tensor_1, np.ndarray) and not isinstance(input_tensor_2, np.ndarray):
|
|
97
|
+
input_tensor_1, input_tensor_2 = input_tensor_2, input_tensor_1
|
|
98
|
+
|
|
95
99
|
disable_strict_mode: bool = kwargs['disable_strict_mode']
|
|
96
100
|
gelu_replace_op_names: dict = kwargs['gelu_replace_op_names']
|
|
97
101
|
|
onnx2tf/ops/ArgMin.py
CHANGED
|
@@ -49,11 +49,15 @@ def make_node(
|
|
|
49
49
|
shape = graph_node_output.shape
|
|
50
50
|
dtype = graph_node_output.dtype
|
|
51
51
|
|
|
52
|
+
# Generation of TF OP
|
|
53
|
+
input_tensor = tf_layers_dict[graph_node_input.name]['tf_node'] \
|
|
54
|
+
if isinstance(graph_node_input, gs.Variable) else graph_node_input
|
|
55
|
+
|
|
52
56
|
axis = graph_node.attrs.get('axis', 0)
|
|
53
57
|
# NCHW->NHWC, NCDHW->NDHWC
|
|
54
58
|
axis = convert_axis(
|
|
55
59
|
axis=axis,
|
|
56
|
-
tensor_rank=len(graph_node_input.shape),
|
|
60
|
+
tensor_rank=len(graph_node_input.shape) if graph_node_input.shape is not None else len(input_tensor.shape),
|
|
57
61
|
before_op_output_shape_trans=before_op_output_shape_trans,
|
|
58
62
|
)
|
|
59
63
|
|
|
@@ -73,10 +77,6 @@ def make_node(
|
|
|
73
77
|
and 'nhwc' in tf_layers_dict[graph_node_input.name].keys() else False
|
|
74
78
|
}
|
|
75
79
|
|
|
76
|
-
# Generation of TF OP
|
|
77
|
-
input_tensor = tf_layers_dict[graph_node_input.name]['tf_node'] \
|
|
78
|
-
if isinstance(graph_node_input, gs.Variable) else graph_node_input
|
|
79
|
-
|
|
80
80
|
# Pre-process transpose
|
|
81
81
|
input_tensor = pre_process_transpose(
|
|
82
82
|
value_before_transpose=input_tensor,
|
onnx2tf/ops/Mul.py
CHANGED
|
@@ -84,6 +84,10 @@ def make_node(
|
|
|
84
84
|
input_tensor_2 = tf_layers_dict[graph_node_input_2.name]['tf_node'] \
|
|
85
85
|
if isinstance(graph_node_input_2, gs.Variable) else graph_node_input_2
|
|
86
86
|
|
|
87
|
+
# issue: https://github.com/PINTO0309/onnx2tf/issues/698
|
|
88
|
+
if isinstance(input_tensor_1, np.ndarray) and not isinstance(input_tensor_2, np.ndarray):
|
|
89
|
+
input_tensor_1, input_tensor_2 = input_tensor_2, input_tensor_1
|
|
90
|
+
|
|
87
91
|
disable_strict_mode: bool = kwargs['disable_strict_mode']
|
|
88
92
|
gelu_replace_op_names: dict = kwargs['gelu_replace_op_names']
|
|
89
93
|
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.1
|
|
2
2
|
Name: onnx2tf
|
|
3
|
-
Version: 1.25.
|
|
3
|
+
Version: 1.25.15
|
|
4
4
|
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).
|
|
5
5
|
Home-page: https://github.com/PINTO0309/onnx2tf
|
|
6
6
|
Author: Katsuya Hyodo
|
|
@@ -314,7 +314,7 @@ Video speed is adjusted approximately 50 times slower than actual speed.
|
|
|
314
314
|
docker run --rm -it \
|
|
315
315
|
-v `pwd`:/workdir \
|
|
316
316
|
-w /workdir \
|
|
317
|
-
ghcr.io/pinto0309/onnx2tf:1.25.
|
|
317
|
+
ghcr.io/pinto0309/onnx2tf:1.25.15
|
|
318
318
|
|
|
319
319
|
or
|
|
320
320
|
|
|
@@ -322,7 +322,7 @@ Video speed is adjusted approximately 50 times slower than actual speed.
|
|
|
322
322
|
docker run --rm -it \
|
|
323
323
|
-v `pwd`:/workdir \
|
|
324
324
|
-w /workdir \
|
|
325
|
-
docker.io/pinto0309/onnx2tf:1.25.
|
|
325
|
+
docker.io/pinto0309/onnx2tf:1.25.15
|
|
326
326
|
|
|
327
327
|
or
|
|
328
328
|
|
|
@@ -1,13 +1,13 @@
|
|
|
1
|
-
onnx2tf/__init__.py,sha256=
|
|
1
|
+
onnx2tf/__init__.py,sha256=opVL1OZ6Gid56hPg_xSsIoFbi2YhdQYGiyMz8JpmmUI,67
|
|
2
2
|
onnx2tf/__main__.py,sha256=2RSCQ7d4lc6CwD-rlGn9UicPFg-P5du7ZD_yh-kuBEU,57
|
|
3
3
|
onnx2tf/onnx2tf.py,sha256=v7juKPrl_0h9SQyck-Ol4B59QXWSSePVC3AUu3fmvII,122750
|
|
4
4
|
onnx2tf/ops/Abs.py,sha256=V7btmCG_ZvK_qJovUsguq0ZMJ349mhNQ4FHSgzP_Yuo,4029
|
|
5
5
|
onnx2tf/ops/Acos.py,sha256=Fo8YkFKuWq8Fi2xUrBdKcAH1yJ8r5pjSD0wgLttTNdk,4003
|
|
6
6
|
onnx2tf/ops/Acosh.py,sha256=ATQj2cT5JS_mTfXi0kXqJ1yzSZu5J0zHA5VjV3j7uKY,3588
|
|
7
|
-
onnx2tf/ops/Add.py,sha256=
|
|
7
|
+
onnx2tf/ops/Add.py,sha256=JnoEYKLFI2Uh7G1MTOvMxGPLIgIJ5V1VAZGKkVa_XzU,12271
|
|
8
8
|
onnx2tf/ops/And.py,sha256=_ubtWa0r8-60x__pS7MEMil1DfBqxiUsk66yRCYS4KY,4591
|
|
9
9
|
onnx2tf/ops/ArgMax.py,sha256=F3PV4EchYQgH1GATJybVGnmY9sGvZkgxCHbNCue9Qns,7278
|
|
10
|
-
onnx2tf/ops/ArgMin.py,sha256=
|
|
10
|
+
onnx2tf/ops/ArgMin.py,sha256=32r7I8AYLQOKTPOOPX1AZwiPnQfkrFB0Le16vdJ1yBs,4225
|
|
11
11
|
onnx2tf/ops/Asin.py,sha256=2djUjTaOzXM6t4Qb-EEMZY-pm1rJl24cgcrep2i_6aQ,4003
|
|
12
12
|
onnx2tf/ops/Asinh.py,sha256=74ZzTEkpxZY4CGfJT2JJU-SHXYL5KZeUkWY2v7hsMMw,3588
|
|
13
13
|
onnx2tf/ops/Atan.py,sha256=D24XDMxEwXFtJheQAr3V3IWOUOc6Q5M0-b_83bmGGMM,3981
|
|
@@ -93,7 +93,7 @@ onnx2tf/ops/MelWeightMatrix.py,sha256=MyYFUTxz2wFVqNx3Dhlro0ktg9kxtEq8sGFmHICDZs
|
|
|
93
93
|
onnx2tf/ops/Min.py,sha256=dK3i115xYh6NusQtGfswEGYBg9MBc_g-edafLgvq4TQ,3356
|
|
94
94
|
onnx2tf/ops/Mish.py,sha256=LEg5MXBLLIzwxmsudC1zTA_yq7drVY_DMCB8lHBCA-8,3546
|
|
95
95
|
onnx2tf/ops/Mod.py,sha256=K6oH5Q4I5JWh8DFp8T1CSdL4WUJCexYdfqTy5iceJxo,9999
|
|
96
|
-
onnx2tf/ops/Mul.py,sha256=
|
|
96
|
+
onnx2tf/ops/Mul.py,sha256=p75MHWbJSo6jLarFzmfK6oQREar4ntlFGqn-U7MzY8s,15962
|
|
97
97
|
onnx2tf/ops/Multinomial.py,sha256=0HQC76IA3AvRsUx9RS0S__nIfEmPuvIaDfSt8bns4FU,3158
|
|
98
98
|
onnx2tf/ops/Neg.py,sha256=vu2ExVXyGpggAM_DNPeZj9QFeUyqhn5XmJnDlPJFsQU,4219
|
|
99
99
|
onnx2tf/ops/NonMaxSuppression.py,sha256=XdqIAh-KhU9JXRxXigGYe37cw2Fus7wgIteJTcIn_wQ,14153
|
|
@@ -188,10 +188,10 @@ onnx2tf/utils/__init__.py,sha256=E9FM9He68VIASDnYp-OrxvHFVn55GzWqw2OEkCqn1zg,27
|
|
|
188
188
|
onnx2tf/utils/common_functions.py,sha256=gCpAe11EcX-gYOMu68saIpIj20pDmLld-6jybnc_0aY,240691
|
|
189
189
|
onnx2tf/utils/enums.py,sha256=7c5TqetqB07VjyHoxJHfLgtqBqk9ZRyUF33fPOJR1IM,1649
|
|
190
190
|
onnx2tf/utils/logging.py,sha256=yUCmPuJ_XiUItM3sZMcaMO24JErkQy7zZwVTYWAuiKg,1982
|
|
191
|
-
onnx2tf-1.25.
|
|
192
|
-
onnx2tf-1.25.
|
|
193
|
-
onnx2tf-1.25.
|
|
194
|
-
onnx2tf-1.25.
|
|
195
|
-
onnx2tf-1.25.
|
|
196
|
-
onnx2tf-1.25.
|
|
197
|
-
onnx2tf-1.25.
|
|
191
|
+
onnx2tf-1.25.15.dist-info/LICENSE,sha256=5v_Kxihy8i6mzHVl349ikSREaIdsl9YeUnX1KBDLD2w,1070
|
|
192
|
+
onnx2tf-1.25.15.dist-info/LICENSE_onnx-tensorflow,sha256=gK4GtS9S5YcyINu6uuNNWdo-kBClyEM4MFLFGiNTeRM,11231
|
|
193
|
+
onnx2tf-1.25.15.dist-info/METADATA,sha256=a3NRQOOxqxrY48M3FAnq6UiitZ3iXaX_zyLHhd02buQ,146125
|
|
194
|
+
onnx2tf-1.25.15.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
|
|
195
|
+
onnx2tf-1.25.15.dist-info/entry_points.txt,sha256=gDPK8ToCFPKMvm8jr9xrGOkXtORJJVh4736fBEKO5k0,41
|
|
196
|
+
onnx2tf-1.25.15.dist-info/top_level.txt,sha256=WgfPiEy3f6vZ_FOpAIEA2CF3TCx1eYrhGw93Ih6b9Fw,8
|
|
197
|
+
onnx2tf-1.25.15.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|