Simple ONNX operation generator. Simple Operation Generator for ONNX.
Project description
[WIP] sog4onnx
Simple ONNX operation generator. Simple Operation Generator for ONNX.
https://github.com/PINTO0309/simple-onnx-processing-tools
Key concept
- Variable, Constant, Operation and Attribute can be generated externally.
- Allow Opset to be specified externally.
- No check for consistency of Operations within the tool, as new OPs are added frequently and the definitions of existing OPs change with each new version of ONNX's Opset.
- Only one OP can be defined at a time, and the goal is to generate free ONNX graphs using a combination of snc4onnx, sne4onnx, snd4onnx and scs4onnx.
- List of parameters that can be specified: https://github.com/onnx/onnx/blob/main/docs/Operators.md
6. Sample
$ sog4onnx \
--op_type Gemm \
--opset 11 \
--input_variables i1 float32 [1,2,3] \
--input_variables i2 float32 [1,1] \
--input_variables i3 int32 [0] \
--output_variables o1 float32 [1,2,3] \
--attributes "{\"alpha\": 1.0, \"beta\": 1.0, \"broadcast\": 0, \"transA\": 0, \"transB\": 0}" \
--non_verbose
7. Reference
- https://docs.nvidia.com/deeplearning/tensorrt/onnx-graphsurgeon/docs/index.html
- https://github.com/NVIDIA/TensorRT/tree/main/tools/onnx-graphsurgeon
- https://github.com/PINTO0309/sne4onnx
- https://github.com/PINTO0309/snd4onnx
- https://github.com/PINTO0309/snc4onnx
- https://github.com/PINTO0309/scs4onnx
- https://github.com/PINTO0309/PINTO_model_zoo
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
sog4onnx-0.0.1.tar.gz
(5.3 kB
view hashes)