onnxslim 0.1.80__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.
- onnxslim/__init__.py +16 -0
- onnxslim/__main__.py +4 -0
- onnxslim/argparser.py +215 -0
- onnxslim/cli/__init__.py +1 -0
- onnxslim/cli/_main.py +180 -0
- onnxslim/core/__init__.py +219 -0
- onnxslim/core/optimization/__init__.py +146 -0
- onnxslim/core/optimization/dead_node_elimination.py +151 -0
- onnxslim/core/optimization/subexpression_elimination.py +76 -0
- onnxslim/core/optimization/weight_tying.py +59 -0
- onnxslim/core/pattern/__init__.py +249 -0
- onnxslim/core/pattern/elimination/__init__.py +5 -0
- onnxslim/core/pattern/elimination/concat.py +61 -0
- onnxslim/core/pattern/elimination/reshape.py +77 -0
- onnxslim/core/pattern/elimination/reshape_as.py +64 -0
- onnxslim/core/pattern/elimination/slice.py +108 -0
- onnxslim/core/pattern/elimination/unsqueeze.py +92 -0
- onnxslim/core/pattern/fusion/__init__.py +8 -0
- onnxslim/core/pattern/fusion/concat_reshape.py +50 -0
- onnxslim/core/pattern/fusion/convadd.py +70 -0
- onnxslim/core/pattern/fusion/convbn.py +86 -0
- onnxslim/core/pattern/fusion/convmul.py +69 -0
- onnxslim/core/pattern/fusion/gelu.py +47 -0
- onnxslim/core/pattern/fusion/gemm.py +330 -0
- onnxslim/core/pattern/fusion/padconv.py +89 -0
- onnxslim/core/pattern/fusion/reduce.py +67 -0
- onnxslim/core/pattern/registry.py +28 -0
- onnxslim/misc/__init__.py +0 -0
- onnxslim/misc/tabulate.py +2681 -0
- onnxslim/third_party/__init__.py +0 -0
- onnxslim/third_party/_sympy/__init__.py +0 -0
- onnxslim/third_party/_sympy/functions.py +205 -0
- onnxslim/third_party/_sympy/numbers.py +397 -0
- onnxslim/third_party/_sympy/printers.py +491 -0
- onnxslim/third_party/_sympy/solve.py +172 -0
- onnxslim/third_party/_sympy/symbol.py +102 -0
- onnxslim/third_party/onnx_graphsurgeon/__init__.py +15 -0
- onnxslim/third_party/onnx_graphsurgeon/exporters/__init__.py +1 -0
- onnxslim/third_party/onnx_graphsurgeon/exporters/base_exporter.py +33 -0
- onnxslim/third_party/onnx_graphsurgeon/exporters/onnx_exporter.py +432 -0
- onnxslim/third_party/onnx_graphsurgeon/graph_pattern/__init__.py +4 -0
- onnxslim/third_party/onnx_graphsurgeon/graph_pattern/graph_pattern.py +466 -0
- onnxslim/third_party/onnx_graphsurgeon/importers/__init__.py +1 -0
- onnxslim/third_party/onnx_graphsurgeon/importers/base_importer.py +33 -0
- onnxslim/third_party/onnx_graphsurgeon/importers/onnx_importer.py +558 -0
- onnxslim/third_party/onnx_graphsurgeon/ir/__init__.py +0 -0
- onnxslim/third_party/onnx_graphsurgeon/ir/function.py +274 -0
- onnxslim/third_party/onnx_graphsurgeon/ir/graph.py +1575 -0
- onnxslim/third_party/onnx_graphsurgeon/ir/node.py +266 -0
- onnxslim/third_party/onnx_graphsurgeon/ir/tensor.py +504 -0
- onnxslim/third_party/onnx_graphsurgeon/logger/__init__.py +1 -0
- onnxslim/third_party/onnx_graphsurgeon/logger/logger.py +261 -0
- onnxslim/third_party/onnx_graphsurgeon/util/__init__.py +0 -0
- onnxslim/third_party/onnx_graphsurgeon/util/exception.py +20 -0
- onnxslim/third_party/onnx_graphsurgeon/util/misc.py +252 -0
- onnxslim/third_party/symbolic_shape_infer.py +3273 -0
- onnxslim/utils.py +794 -0
- onnxslim/version.py +1 -0
- onnxslim-0.1.80.dist-info/METADATA +207 -0
- onnxslim-0.1.80.dist-info/RECORD +65 -0
- onnxslim-0.1.80.dist-info/WHEEL +5 -0
- onnxslim-0.1.80.dist-info/entry_points.txt +2 -0
- onnxslim-0.1.80.dist-info/licenses/LICENSE +21 -0
- onnxslim-0.1.80.dist-info/top_level.txt +1 -0
- onnxslim-0.1.80.dist-info/zip-safe +1 -0
onnxslim/version.py
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
__version__ = "0.1.80"
|
|
@@ -0,0 +1,207 @@
|
|
|
1
|
+
Metadata-Version: 2.4
|
|
2
|
+
Name: onnxslim
|
|
3
|
+
Version: 0.1.80
|
|
4
|
+
Summary: OnnxSlim: A Toolkit to Help Optimize Onnx Model
|
|
5
|
+
Home-page: https://github.com/inisis/OnnxSlim
|
|
6
|
+
Author: inisis
|
|
7
|
+
Author-email: desmond.yao@buaa.edu.cn
|
|
8
|
+
License: MIT
|
|
9
|
+
Project-URL: Bug Tracker, https://github.com/inisis/OnnxSlim/issues
|
|
10
|
+
Classifier: Programming Language :: Python :: 3
|
|
11
|
+
Classifier: License :: OSI Approved :: MIT License
|
|
12
|
+
Classifier: Intended Audience :: Developers
|
|
13
|
+
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
|
|
14
|
+
Requires-Python: >=3.6
|
|
15
|
+
Description-Content-Type: text/markdown
|
|
16
|
+
License-File: LICENSE
|
|
17
|
+
Requires-Dist: onnx
|
|
18
|
+
Requires-Dist: sympy>=1.13.1
|
|
19
|
+
Requires-Dist: packaging
|
|
20
|
+
Requires-Dist: colorama
|
|
21
|
+
Requires-Dist: ml_dtypes
|
|
22
|
+
Dynamic: author
|
|
23
|
+
Dynamic: author-email
|
|
24
|
+
Dynamic: classifier
|
|
25
|
+
Dynamic: description
|
|
26
|
+
Dynamic: description-content-type
|
|
27
|
+
Dynamic: home-page
|
|
28
|
+
Dynamic: license
|
|
29
|
+
Dynamic: license-file
|
|
30
|
+
Dynamic: project-url
|
|
31
|
+
Dynamic: requires-dist
|
|
32
|
+
Dynamic: requires-python
|
|
33
|
+
Dynamic: summary
|
|
34
|
+
|
|
35
|
+
# OnnxSlim
|
|
36
|
+
|
|
37
|
+
<p align="center">
|
|
38
|
+
<a href="https://pypi.org/project/onnxslim">
|
|
39
|
+
<img src="https://img.shields.io/pypi/v/onnxslim?color=blue" />
|
|
40
|
+
</a>
|
|
41
|
+
<a href="https://pypi.org/project/onnxslim">
|
|
42
|
+
<img src="https://static.pepy.tech/badge/onnxslim/week" />
|
|
43
|
+
</a>
|
|
44
|
+
<a href="https://pypi.org/project/onnxslim">
|
|
45
|
+
<img src="https://static.pepy.tech/badge/onnxslim/month" />
|
|
46
|
+
</a>
|
|
47
|
+
<a href="https://pypi.org/project/onnxslim">
|
|
48
|
+
<img src="https://static.pepy.tech/badge/onnxslim" />
|
|
49
|
+
</a>
|
|
50
|
+
<a href="https://github.com/inisis/onnxslim/actions/workflows/ci.yaml">
|
|
51
|
+
<img src="https://github.com/inisis/onnxslim/actions/workflows/ci.yml/badge.svg" />
|
|
52
|
+
</a>
|
|
53
|
+
<a href="https://codecov.io/gh/inisis/onnxslim" >
|
|
54
|
+
<img src="https://codecov.io/gh/inisis/onnxslim/branch/main/graph/badge.svg?token=C69ZH6802N"/>
|
|
55
|
+
</a>
|
|
56
|
+
<a href="https://muhammadrizwanmunawar.medium.com/boost-onnx-load-speed-by-10-15-with-onnxslims-python-package-d401eb8c2e69">
|
|
57
|
+
<img src="https://img.shields.io/badge/Blog-OnnxSlim?style=flat&label=OnnxSlim" />
|
|
58
|
+
</a>
|
|
59
|
+
<a href="https://deepwiki.com/inisis/OnnxSlim"><img src="https://img.shields.io/badge/DeepWiki-inisis%2FOnnxSlim-blue.svg?logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAACwAAAAyCAYAAAAnWDnqAAAAAXNSR0IArs4c6QAAA05JREFUaEPtmUtyEzEQhtWTQyQLHNak2AB7ZnyXZMEjXMGeK/AIi+QuHrMnbChYY7MIh8g01fJoopFb0uhhEqqcbWTp06/uv1saEDv4O3n3dV60RfP947Mm9/SQc0ICFQgzfc4CYZoTPAswgSJCCUJUnAAoRHOAUOcATwbmVLWdGoH//PB8mnKqScAhsD0kYP3j/Yt5LPQe2KvcXmGvRHcDnpxfL2zOYJ1mFwrryWTz0advv1Ut4CJgf5uhDuDj5eUcAUoahrdY/56ebRWeraTjMt/00Sh3UDtjgHtQNHwcRGOC98BJEAEymycmYcWwOprTgcB6VZ5JK5TAJ+fXGLBm3FDAmn6oPPjR4rKCAoJCal2eAiQp2x0vxTPB3ALO2CRkwmDy5WohzBDwSEFKRwPbknEggCPB/imwrycgxX2NzoMCHhPkDwqYMr9tRcP5qNrMZHkVnOjRMWwLCcr8ohBVb1OMjxLwGCvjTikrsBOiA6fNyCrm8V1rP93iVPpwaE+gO0SsWmPiXB+jikdf6SizrT5qKasx5j8ABbHpFTx+vFXp9EnYQmLx02h1QTTrl6eDqxLnGjporxl3NL3agEvXdT0WmEost648sQOYAeJS9Q7bfUVoMGnjo4AZdUMQku50McDcMWcBPvr0SzbTAFDfvJqwLzgxwATnCgnp4wDl6Aa+Ax283gghmj+vj7feE2KBBRMW3FzOpLOADl0Isb5587h/U4gGvkt5v60Z1VLG8BhYjbzRwyQZemwAd6cCR5/XFWLYZRIMpX39AR0tjaGGiGzLVyhse5C9RKC6ai42ppWPKiBagOvaYk8lO7DajerabOZP46Lby5wKjw1HCRx7p9sVMOWGzb/vA1hwiWc6jm3MvQDTogQkiqIhJV0nBQBTU+3okKCFDy9WwferkHjtxib7t3xIUQtHxnIwtx4mpg26/HfwVNVDb4oI9RHmx5WGelRVlrtiw43zboCLaxv46AZeB3IlTkwouebTr1y2NjSpHz68WNFjHvupy3q8TFn3Hos2IAk4Ju5dCo8B3wP7VPr/FGaKiG+T+v+TQqIrOqMTL1VdWV1DdmcbO8KXBz6esmYWYKPwDL5b5FA1a0hwapHiom0r/cKaoqr+27/XcrS5UwSMbQAAAABJRU5ErkJggg==" alt="DeepWiki"></a>
|
|
60
|
+
</p>
|
|
61
|
+
|
|
62
|
+
OnnxSlim can help you slim your onnx model, with less operators, but same accuracy, better inference speed.
|
|
63
|
+
|
|
64
|
+
- 🚀 2025/11/29: Top 1% on PyPI
|
|
65
|
+
- 🚀 2025/11/27: OnnxSlim is merged into [NVIDIA TensorRT-Model-Optimizer](https://github.com/NVIDIA/TensorRT-Model-Optimizer) 🤗🤗🤗
|
|
66
|
+
- 🚀 2025/05/17: OnnxSlim is merged into [HuggingFace optimum](https://github.com/huggingface/optimum) 🤗🤗🤗
|
|
67
|
+
- 🚀 2025/04/30: Rank 1st in the [AICAS 2025 LLM inference optimization challenge](https://tianchi.aliyun.com/competition/entrance/532289/customize588)
|
|
68
|
+
- 🚀 2025/01/28: Achieved 1M downloads
|
|
69
|
+
- 🚀 2024/06/23: OnnxSlim is merged into [transformers.js](https://github.com/huggingface/transformers.js) 🤗🤗🤗
|
|
70
|
+
- 🚀 2024/06/02: OnnxSlim is merged into [ultralytics](https://github.com/ultralytics/ultralytics) ❤️❤️❤️
|
|
71
|
+
- 🚀 2024/04/30: Rank 1st in the [AICAS 2024 LLM inference optimization challenge](https://tianchi.aliyun.com/competition/entrance/532170/customize440) held by Arm and T-head
|
|
72
|
+
- 🚀 2024/01/25: OnnxSlim is merged to [mnn-llm](https://github.com/wangzhaode/mnn-llm), performance increased by 5%
|
|
73
|
+
|
|
74
|
+
# Benchmark
|
|
75
|
+
|
|
76
|
+

|
|
77
|
+
|
|
78
|
+
# Installation
|
|
79
|
+
|
|
80
|
+
## Using Prebuilt
|
|
81
|
+
|
|
82
|
+
```bash
|
|
83
|
+
pip install onnxslim
|
|
84
|
+
```
|
|
85
|
+
|
|
86
|
+
## Install From Source
|
|
87
|
+
|
|
88
|
+
```bash
|
|
89
|
+
pip install git+https://github.com/inisis/OnnxSlim@main
|
|
90
|
+
```
|
|
91
|
+
|
|
92
|
+
## Install From Local
|
|
93
|
+
|
|
94
|
+
```bash
|
|
95
|
+
git clone https://github.com/inisis/OnnxSlim && cd OnnxSlim/
|
|
96
|
+
pip install .
|
|
97
|
+
```
|
|
98
|
+
|
|
99
|
+
# How to use
|
|
100
|
+
|
|
101
|
+
## Bash
|
|
102
|
+
|
|
103
|
+
```bash
|
|
104
|
+
onnxslim your_onnx_model slimmed_onnx_model
|
|
105
|
+
```
|
|
106
|
+
|
|
107
|
+
<div align=left><img src="https://raw.githubusercontent.com/inisis/onnxslim/main/images/onnxslim.gif"></div>
|
|
108
|
+
|
|
109
|
+
## Inscript
|
|
110
|
+
|
|
111
|
+
```inscript
|
|
112
|
+
import onnx
|
|
113
|
+
import onnxslim
|
|
114
|
+
|
|
115
|
+
model = onnx.load("model.onnx")
|
|
116
|
+
slimmed_model = onnxslim.slim(model)
|
|
117
|
+
onnx.save(slimmed_model, "slimmed_model.onnx")
|
|
118
|
+
```
|
|
119
|
+
|
|
120
|
+
For more usage, see onnxslim -h or refer to our [examples](./examples)
|
|
121
|
+
|
|
122
|
+
# Projects using OnnxSlim
|
|
123
|
+
|
|
124
|
+
<table style="width:100%; border-collapse:separate; border-spacing:10px;">
|
|
125
|
+
<tr>
|
|
126
|
+
<td style="vertical-align:middle;">
|
|
127
|
+
<img src="https://avatars.githubusercontent.com/u/1728152?s=200&v=4" width="22" height="22" style="vertical-align:middle; margin-right:8px;"/>
|
|
128
|
+
<a href="https://github.com/NVIDIA/TensorRT-Model-Optimizer" target="_blank">NVIDIA/TensorRT-Model-Optimizer</a>
|
|
129
|
+
</td>
|
|
130
|
+
<td style="vertical-align:middle;">
|
|
131
|
+
<img src="https://avatars.githubusercontent.com/u/1961952?s=48&v=4" width="22" height="22" style="vertical-align:middle; margin-right:8px;"/>
|
|
132
|
+
<a href="https://github.com/alibaba/MNN" target="_blank">alibaba/MNN</a>
|
|
133
|
+
</td>
|
|
134
|
+
</tr>
|
|
135
|
+
<tr>
|
|
136
|
+
<td style="vertical-align:middle;">
|
|
137
|
+
<img src="https://avatars.githubusercontent.com/u/26833451?s=48&v=4" width="22" height="22" style="vertical-align:middle; margin-right:8px;"/>
|
|
138
|
+
<a href="https://github.com/ultralytics/ultralytics" target="_blank">ultralytics/ultralytics</a>
|
|
139
|
+
</td>
|
|
140
|
+
<td style="vertical-align:middle;">
|
|
141
|
+
<img src="https://avatars.githubusercontent.com/u/131524?s=48&v=4" width="22" height="22" style="vertical-align:middle; margin-right:8px;"/>
|
|
142
|
+
<a href="https://github.com/mozilla/smart_autofill" target="_blank">Mozilla/smart_autofill</a>
|
|
143
|
+
</td>
|
|
144
|
+
</tr>
|
|
145
|
+
<tr>
|
|
146
|
+
<td style="vertical-align:middle;">
|
|
147
|
+
<img src="https://avatars.githubusercontent.com/u/1961952?s=48&v=4" width="22" height="22" style="vertical-align:middle; margin-right:8px;"/>
|
|
148
|
+
<a href="https://github.com/wangzhaode/mnn-llm" target="_blank">alibaba/MNN-LLM</a>
|
|
149
|
+
</td>
|
|
150
|
+
<td style="vertical-align:middle;">
|
|
151
|
+
<img src="https://avatars.githubusercontent.com/u/25720743?s=48&v=4" width="22" height="22" style="vertical-align:middle; margin-right:8px;"/>
|
|
152
|
+
<a href="https://github.com/huggingface/transformers.js" target="_blank">huggingface/transformers.js</a>
|
|
153
|
+
</td>
|
|
154
|
+
</tr>
|
|
155
|
+
<tr>
|
|
156
|
+
<td style="vertical-align:middle;">
|
|
157
|
+
<img src="https://avatars.githubusercontent.com/u/25720743?s=48&v=4" width="22" height="22" style="vertical-align:middle; margin-right:8px;"/>
|
|
158
|
+
<a href="https://github.com/huggingface/optimum" target="_blank">huggingface/optimum</a>
|
|
159
|
+
</td>
|
|
160
|
+
<td style="vertical-align:middle;">
|
|
161
|
+
<img src="https://avatars.githubusercontent.com/u/23534030?s=48&v=4" width="22" height="22" style="vertical-align:middle; margin-right:8px;"/>
|
|
162
|
+
<a href="https://github.com/PaddlePaddle/PaddleOCR" target="_blank">PaddlePaddle/PaddleOCR</a>
|
|
163
|
+
</td>
|
|
164
|
+
</tr>
|
|
165
|
+
<tr>
|
|
166
|
+
<td style="vertical-align:middle;">
|
|
167
|
+
<img src="https://avatars.githubusercontent.com/u/109945100?s=48&v=4" width="22" height="22" style="vertical-align:middle; margin-right:8px;"/>
|
|
168
|
+
<a href="https://github.com/modelscope/FunASR" target="_blank">ModelScope/FunASR</a>
|
|
169
|
+
</td>
|
|
170
|
+
<td style="vertical-align:middle;">
|
|
171
|
+
<img src="https://avatars.githubusercontent.com/u/111754012?s=48&v=4" width="22" height="22" style="vertical-align:middle; margin-right:8px;"/>
|
|
172
|
+
<a href="https://github.com/CVCUDA/CV-CUDA" target="_blank">CVCUDA/CV-CUDA</a>
|
|
173
|
+
</td>
|
|
174
|
+
</tr>
|
|
175
|
+
<tr>
|
|
176
|
+
<td style="vertical-align:middle;">
|
|
177
|
+
<img src="https://avatars.githubusercontent.com/u/86091366?s=48&v=4" width="22" height="22" style="vertical-align:middle; margin-right:8px;"/>
|
|
178
|
+
<a href="https://github.com/THU-MIG/yolov10" target="_blank">THU-MIG/yolov10</a>
|
|
179
|
+
</td>
|
|
180
|
+
<td style="vertical-align:middle;">
|
|
181
|
+
<img src="https://avatars.githubusercontent.com/u/48153283?s=48&v=4" width="22" height="22" style="vertical-align:middle; margin-right:8px;"/>
|
|
182
|
+
<a href="https://github.com/sunsmarterjie/yolov12" target="_blank">sunsmarterjie/yolov12</a>
|
|
183
|
+
</td>
|
|
184
|
+
</tr>
|
|
185
|
+
<tr>
|
|
186
|
+
<td style="vertical-align:middle;">
|
|
187
|
+
<img src="https://avatars.githubusercontent.com/u/147458884?s=48&v=4" width="22" height="22" style="vertical-align:middle; margin-right:8px;"/>
|
|
188
|
+
<a href="https://github.com/nndeploy/nndeploy" target="_blank">nndeploy/nndeploy</a>
|
|
189
|
+
</td>
|
|
190
|
+
<td style="vertical-align:middle;">
|
|
191
|
+
<img src="https://avatars.githubusercontent.com/u/126587470?s=48&v=4" width="22" height="22" style="vertical-align:middle; margin-right:8px;"/>
|
|
192
|
+
<a href="https://github.com/deepghs/imgutils" target="_blank">deepghs/imgutils</a>
|
|
193
|
+
</td>
|
|
194
|
+
</tr>
|
|
195
|
+
</table>
|
|
196
|
+
|
|
197
|
+
# References
|
|
198
|
+
|
|
199
|
+
> - [onnx-graphsurgeon](https://github.com/NVIDIA/TensorRT/tree/main/tools/onnx-graphsurgeon)
|
|
200
|
+
> - [Polygraphy](https://github.com/NVIDIA/TensorRT/tree/main/tools/Polygraphy/polygraphy)
|
|
201
|
+
> - [onnx-simplifier](https://github.com/daquexian/onnx-simplifier)
|
|
202
|
+
> - [tabulate](https://github.com/astanin/python-tabulate)
|
|
203
|
+
> - [onnxruntime](https://github.com/microsoft/onnxruntime)
|
|
204
|
+
|
|
205
|
+
# Contact
|
|
206
|
+
|
|
207
|
+
Discord: https://discord.gg/nRw2Fd3VUS QQ Group: `873569894`
|
|
@@ -0,0 +1,65 @@
|
|
|
1
|
+
onnxslim/__init__.py,sha256=ECHGdxzg4b-4SZaPhxM_KulBi-xDbVcVUbpJc8i6a60,571
|
|
2
|
+
onnxslim/__main__.py,sha256=FgDcl6xX8kV_52rB-jPVsmGqidlVhkpe_YhXK75-nFU,75
|
|
3
|
+
onnxslim/argparser.py,sha256=pFv3nEZH2BiHO9ejS4Iq5ZuZ3GrpdyRQJypAyR0xF7w,8942
|
|
4
|
+
onnxslim/utils.py,sha256=Z39vRKAtD7o18NFbf3Qrws9xDg81uBJ9F-RYEFAfqM8,28095
|
|
5
|
+
onnxslim/version.py,sha256=7BO6MGkPGuN-ZrZBKBek9x0F-EwGEiaELuB5k1018oE,23
|
|
6
|
+
onnxslim/cli/__init__.py,sha256=kxK27cDgWotBOWRs86rbRQf_dtmniKr1GZJeasxfESE,42
|
|
7
|
+
onnxslim/cli/_main.py,sha256=jEKv9q7y_y9g0GsrfXcnk_wyMVej6jhe9QNPChE7yTs,5550
|
|
8
|
+
onnxslim/core/__init__.py,sha256=uDg-Eu29Ezb3txwZf5mN0zQRVuqF-K9BvktE8WBYS4E,8825
|
|
9
|
+
onnxslim/core/optimization/__init__.py,sha256=6w7VEhsfd7MhCSLmpAMLfxaEq9pf4YYHhNndggNLZCg,5034
|
|
10
|
+
onnxslim/core/optimization/dead_node_elimination.py,sha256=JoQsWb304NSqjssFTvlbOog1N-2KLitZ-alrfex3Q88,7558
|
|
11
|
+
onnxslim/core/optimization/subexpression_elimination.py,sha256=wauPADo0msVifYGgJ_fiNmPSe7LrJ7A-D3ur5z1wif8,2838
|
|
12
|
+
onnxslim/core/optimization/weight_tying.py,sha256=uAz_2AJa-s-budTcDXIr6ZcME1hXPdm54CzR-u850X4,2347
|
|
13
|
+
onnxslim/core/pattern/__init__.py,sha256=az4HayGp9WybHMiijr-VfbhzPIfRc-51JK8J1_Q-5Yo,9953
|
|
14
|
+
onnxslim/core/pattern/registry.py,sha256=dMW5fisu_traJ3sYPoy_NU83S1xfGjqxV2GbB0OSYNg,874
|
|
15
|
+
onnxslim/core/pattern/elimination/__init__.py,sha256=C9EwJj7DQmaXVvGx6wxvqvCdQGEAAYVQu1zWQ3PCB5E,117
|
|
16
|
+
onnxslim/core/pattern/elimination/concat.py,sha256=RmN3B0qtVixE_7QfgxsJHj2MUPOEdp8oxrcFN2oSR5Q,2261
|
|
17
|
+
onnxslim/core/pattern/elimination/reshape.py,sha256=XwvuPAZnXCCEwJb2n1guigstnsl3wlxGygytH3GZXN8,3109
|
|
18
|
+
onnxslim/core/pattern/elimination/reshape_as.py,sha256=FI3LYR0pzbp2pDmaX13duHrQ4uqwaKNu4bG78en-7wY,2034
|
|
19
|
+
onnxslim/core/pattern/elimination/slice.py,sha256=moZibU-TbtdwtmGIUwyjnjf3oRCeCBcQq0M1gY5ZWDk,5033
|
|
20
|
+
onnxslim/core/pattern/elimination/unsqueeze.py,sha256=v7Rin3qB6F49ETrxXWEQQxUgtlF18nvHb6JFarf0kwQ,3855
|
|
21
|
+
onnxslim/core/pattern/fusion/__init__.py,sha256=3ajHvRurL7WHL4tfNsBoLQh6Sq2fyiqH-VsPuftYMGg,183
|
|
22
|
+
onnxslim/core/pattern/fusion/concat_reshape.py,sha256=LvknixTAsSUqUkGSuoEA1QpC-TmBrsx6AHZoeT0gTbI,1615
|
|
23
|
+
onnxslim/core/pattern/fusion/convadd.py,sha256=P1GI7hJAHgDBO17aDDghNxMEhWkFIcqGLIfnpTMGhWk,2432
|
|
24
|
+
onnxslim/core/pattern/fusion/convbn.py,sha256=696Ev3X9-G2eQbnSqpqkQQeWG8Lrb-3elpQiOQ6MAXE,3333
|
|
25
|
+
onnxslim/core/pattern/fusion/convmul.py,sha256=W2C6H3kWSDUg0he0jfR4tXI5GMi7gsyylQR4aSh-rik,2581
|
|
26
|
+
onnxslim/core/pattern/fusion/gelu.py,sha256=uR67AJ_tL1gboY6VsTdqajHxW3Pbu656UMhCe1mQZDY,1469
|
|
27
|
+
onnxslim/core/pattern/fusion/gemm.py,sha256=Ti9yZAfEprFRvW1FiAD0zvewELOJbRjposIk3yjjXfQ,12928
|
|
28
|
+
onnxslim/core/pattern/fusion/padconv.py,sha256=eOutev5rOrHuyyw-BRIFzMjcvu9MxXj73kY215GaeG8,3652
|
|
29
|
+
onnxslim/core/pattern/fusion/reduce.py,sha256=dMC7CPlFglrJxugsJWjcc-jQCIa_GIbW1y9K2FRvvcE,2755
|
|
30
|
+
onnxslim/misc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
31
|
+
onnxslim/misc/tabulate.py,sha256=Pg5uU0UP18HbwG-c8LlA82LbIb_5JWQeuIB1AnturbM,99695
|
|
32
|
+
onnxslim/third_party/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
33
|
+
onnxslim/third_party/symbolic_shape_infer.py,sha256=pCuHkqXVH0yF5e9GUTwrs01-GoXEGOoy7w_J1DsbLW8,152797
|
|
34
|
+
onnxslim/third_party/_sympy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
35
|
+
onnxslim/third_party/_sympy/functions.py,sha256=s3pKzyjYCKnvlddLFR_H8UmbbcdMB51PRxqhe9zGI9E,8876
|
|
36
|
+
onnxslim/third_party/_sympy/numbers.py,sha256=WOCwRg5JiIfUQCXKAUUuWis8rxdc7LLthUf3Ax1jG8I,11535
|
|
37
|
+
onnxslim/third_party/_sympy/printers.py,sha256=Kv2vpR-YgjCsNxIBKkcHyvEnj_H-gmJqG03Hwh4rWdk,20429
|
|
38
|
+
onnxslim/third_party/_sympy/solve.py,sha256=gcqmluQbAKzIQTTtzsgALguhK8ViRGlVP-CRDcbwP6A,6465
|
|
39
|
+
onnxslim/third_party/_sympy/symbol.py,sha256=aAh52bmsGzRzG3PqS8MNcs3rjTil7Mqt4x3dev6fnKc,3718
|
|
40
|
+
onnxslim/third_party/onnx_graphsurgeon/__init__.py,sha256=hgX680FHJKYEqlgCexdbdM8LfffxDD_GYFEnI7jnE1I,695
|
|
41
|
+
onnxslim/third_party/onnx_graphsurgeon/exporters/__init__.py,sha256=coVGPk31uciLYrpJmsDPypTAXqN7V1GgyNCXPp59KiQ,88
|
|
42
|
+
onnxslim/third_party/onnx_graphsurgeon/exporters/base_exporter.py,sha256=I2dKwwj4m0lsEhR0ngH-huBvsLcL9jh7o2RQGwMz2ik,1150
|
|
43
|
+
onnxslim/third_party/onnx_graphsurgeon/exporters/onnx_exporter.py,sha256=JBbY7bl1b39Hk9j9w2FzC5WZ3NKH6BJR3PyXGT-HTdQ,17551
|
|
44
|
+
onnxslim/third_party/onnx_graphsurgeon/graph_pattern/__init__.py,sha256=4r6nr2YY9eJrJG18Ae-kdW_hSOuNSMFBSYygmS9GT9U,121
|
|
45
|
+
onnxslim/third_party/onnx_graphsurgeon/graph_pattern/graph_pattern.py,sha256=Nn5lkO4JKFRhaF4EPj3xuG15fSt8zvLfRI3w_7XmL_g,20053
|
|
46
|
+
onnxslim/third_party/onnx_graphsurgeon/importers/__init__.py,sha256=ON_tO_sODloL39hhGjE19sz_Cj7KeGnsqPh-JlOPc4k,88
|
|
47
|
+
onnxslim/third_party/onnx_graphsurgeon/importers/base_importer.py,sha256=ESIul1po3LWkNz8D22Ti_KFMpBe7FQF3iKsh32yABXg,1170
|
|
48
|
+
onnxslim/third_party/onnx_graphsurgeon/importers/onnx_importer.py,sha256=qa86Ne8yWCmpoAPBWV2lV1hlCvnQ6UPe-M1JXSfnMqM,23097
|
|
49
|
+
onnxslim/third_party/onnx_graphsurgeon/ir/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
50
|
+
onnxslim/third_party/onnx_graphsurgeon/ir/function.py,sha256=X1Rd1ZQlHhK6crg788a-LCmQSzv446LGfw376_Cz8Co,11820
|
|
51
|
+
onnxslim/third_party/onnx_graphsurgeon/ir/graph.py,sha256=RU1luTR5sGMbPbRXtKGsYtBIv5BlXZOo7gU6bv0L5FY,70494
|
|
52
|
+
onnxslim/third_party/onnx_graphsurgeon/ir/node.py,sha256=lHrJCNRhtPRZrE7vuvQkG_wfEsJzDW7Wf-T_kr4OJHI,9996
|
|
53
|
+
onnxslim/third_party/onnx_graphsurgeon/ir/tensor.py,sha256=bypjlsVp1qByPhJRbTSjSrPpoatmMykjnJ9_cnnmz9Y,19265
|
|
54
|
+
onnxslim/third_party/onnx_graphsurgeon/logger/__init__.py,sha256=b6lAvvrKZKNtCZOgcvz2Aj9lUO5mw5JM8UFP5BqBOnQ,83
|
|
55
|
+
onnxslim/third_party/onnx_graphsurgeon/logger/logger.py,sha256=L12rrwn33RHH-2WLvRwN77CyHezK1DM7AE4RQ2v_3-Y,10350
|
|
56
|
+
onnxslim/third_party/onnx_graphsurgeon/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
|
57
|
+
onnxslim/third_party/onnx_graphsurgeon/util/exception.py,sha256=KrsHbKEQ4237UbjlODsUzvkXoAY72LZi23ApBeFANWg,786
|
|
58
|
+
onnxslim/third_party/onnx_graphsurgeon/util/misc.py,sha256=kyxInD2SCRLU4wHMeiDEYEHB3871fGks6kQTuF9uATY,8960
|
|
59
|
+
onnxslim-0.1.80.dist-info/licenses/LICENSE,sha256=oHZXw-yrBwdNVGu4JtlZhMgmQHKIZ7BJJlJdhu1HKvI,1062
|
|
60
|
+
onnxslim-0.1.80.dist-info/METADATA,sha256=WySwYSuyzeAOmlPUtOoyHE0WmbLPXNtyBelpCBYBIqo,10247
|
|
61
|
+
onnxslim-0.1.80.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
|
62
|
+
onnxslim-0.1.80.dist-info/entry_points.txt,sha256=O2QgceCVeGeRhnxRSDRcGiFd0ZNfElwrTiRo1W2V7KA,47
|
|
63
|
+
onnxslim-0.1.80.dist-info/top_level.txt,sha256=EWFTb99i0kc6cC9akqNKp88ipzg17_VZzYN7z1kQNlA,9
|
|
64
|
+
onnxslim-0.1.80.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
|
|
65
|
+
onnxslim-0.1.80.dist-info/RECORD,,
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
MIT License
|
|
2
|
+
|
|
3
|
+
Copyright (c) 2022 inisis
|
|
4
|
+
|
|
5
|
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
6
|
+
of this software and associated documentation files (the "Software"), to deal
|
|
7
|
+
in the Software without restriction, including without limitation the rights
|
|
8
|
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
9
|
+
copies of the Software, and to permit persons to whom the Software is
|
|
10
|
+
furnished to do so, subject to the following conditions:
|
|
11
|
+
|
|
12
|
+
The above copyright notice and this permission notice shall be included in all
|
|
13
|
+
copies or substantial portions of the Software.
|
|
14
|
+
|
|
15
|
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
16
|
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
17
|
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
18
|
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
19
|
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
20
|
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
21
|
+
SOFTWARE.
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
onnxslim
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
|