onnxslim 0.1.81__py3-none-any.whl → 0.1.82__py3-none-any.whl

onnxslim/core/optimization/dead_node_elimination.py

@@ -53,10 +53,10 @@ def dead_node_elimination(graph, is_subgraph=False):
                     node.inputs.pop(1)
                     node.inputs.insert(1, reshape_const)
                     logger.debug(f"replacing {node.op} op: {node.name}")
-        elif node.op == "Slice":
-            if node.inputs[0].shape and node.outputs[0].shape and node.inputs[0].shape == node.outputs[0].shape:
-                node.erase()
-                logger.debug(f"removing {node.op} op: {node.name}")
+        # elif node.op == "Slice":
+        #     if node.inputs[0].shape and node.outputs[0].shape and node.inputs[0].shape == node.outputs[0].shape and all(isinstance(item, int) for item in node.inputs[0].shape):
+        #         node.erase()
+        #         logger.debug(f"removing {node.op} op: {node.name}")
         elif node.op == "Mul":
             if (isinstance(node.inputs[1], Constant) and isinstance(node.inputs[0], Variable)) or (
                 isinstance(node.inputs[0], Constant) and isinstance(node.inputs[1], Variable)

{onnxslim-0.1.81.dist-info → onnxslim-0.1.82.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: onnxslim
-Version: 0.1.81
+Version: 0.1.82
 Summary: OnnxSlim: A Toolkit to Help Optimize Onnx Model
 Project-URL: homepage, https://github.com/inisis/OnnxSlim
 Project-URL: issues, https://github.com/inisis/OnnxSlim/issues

{onnxslim-0.1.81.dist-info → onnxslim-0.1.82.dist-info}/RECORD

@@ -7,7 +7,7 @@ onnxslim/cli/__init__.py,sha256=kxK27cDgWotBOWRs86rbRQf_dtmniKr1GZJeasxfESE,42
 onnxslim/cli/_main.py,sha256=jEKv9q7y_y9g0GsrfXcnk_wyMVej6jhe9QNPChE7yTs,5550
 onnxslim/core/__init__.py,sha256=uDg-Eu29Ezb3txwZf5mN0zQRVuqF-K9BvktE8WBYS4E,8825
 onnxslim/core/optimization/__init__.py,sha256=6w7VEhsfd7MhCSLmpAMLfxaEq9pf4YYHhNndggNLZCg,5034
-onnxslim/core/optimization/dead_node_elimination.py,sha256=B5hpxwLrdWD5uFt_r87uOgSX-Ul81-sNJlYowa4jU_Y,7801
+onnxslim/core/optimization/dead_node_elimination.py,sha256=CVnSq_D_tYpziMmD8SO0IkmhvwpgZDQL0H-Cfe2zK-I,7873
 onnxslim/core/optimization/subexpression_elimination.py,sha256=wauPADo0msVifYGgJ_fiNmPSe7LrJ7A-D3ur5z1wif8,2838
 onnxslim/core/optimization/weight_tying.py,sha256=uAz_2AJa-s-budTcDXIr6ZcME1hXPdm54CzR-u850X4,2347
 onnxslim/core/pattern/__init__.py,sha256=az4HayGp9WybHMiijr-VfbhzPIfRc-51JK8J1_Q-5Yo,9953
@@ -56,8 +56,8 @@ onnxslim/third_party/onnx_graphsurgeon/logger/logger.py,sha256=L12rrwn33RHH-2WLv
 onnxslim/third_party/onnx_graphsurgeon/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 onnxslim/third_party/onnx_graphsurgeon/util/exception.py,sha256=KrsHbKEQ4237UbjlODsUzvkXoAY72LZi23ApBeFANWg,786
 onnxslim/third_party/onnx_graphsurgeon/util/misc.py,sha256=kyxInD2SCRLU4wHMeiDEYEHB3871fGks6kQTuF9uATY,8960
-onnxslim-0.1.81.dist-info/METADATA,sha256=zE4GD0vim78Jk_LYDhUxmjX4t3U9m4eewdZUCyQR76c,10216
-onnxslim-0.1.81.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-onnxslim-0.1.81.dist-info/entry_points.txt,sha256=O2QgceCVeGeRhnxRSDRcGiFd0ZNfElwrTiRo1W2V7KA,47
-onnxslim-0.1.81.dist-info/licenses/LICENSE,sha256=oHZXw-yrBwdNVGu4JtlZhMgmQHKIZ7BJJlJdhu1HKvI,1062
-onnxslim-0.1.81.dist-info/RECORD,,
+onnxslim-0.1.82.dist-info/METADATA,sha256=ckdfDDYyoew97VKJt1C7r_Vl-FK_NfQl0rR-Da1E2PE,10216
+onnxslim-0.1.82.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+onnxslim-0.1.82.dist-info/entry_points.txt,sha256=O2QgceCVeGeRhnxRSDRcGiFd0ZNfElwrTiRo1W2V7KA,47
+onnxslim-0.1.82.dist-info/licenses/LICENSE,sha256=oHZXw-yrBwdNVGu4JtlZhMgmQHKIZ7BJJlJdhu1HKvI,1062
+onnxslim-0.1.82.dist-info/RECORD,,