├── .gitignore ├── Cargo.lock ├── Cargo.toml ├── README.md ├── examples └── test_resnet.rs ├── model_file ├── car.jpg ├── resnet18_batch1.pnnx.bin ├── resnet18_batch1.pnnx.param ├── simple_ops.pnnx.bin ├── simple_ops.pnnx.param ├── simple_ops2.pnnx.bin ├── simple_ops2.pnnx.param ├── test_linear.pnnx.bin ├── test_linear.pnnx.param ├── yolov5s.pnnx.bin └── yolov5s.pnnx.param ├── src ├── data │ ├── mod.rs │ ├── tensor.rs │ └── torch_utils.rs ├── layer │ ├── abstract_layer │ │ ├── layer.rs │ │ ├── layer_factory.rs │ │ └── mod.rs │ ├── details │ │ ├── adaptive_avgpooling.rs │ │ ├── convolution.rs │ │ ├── expression.rs │ │ ├── flatten.rs │ │ ├── linear.rs │ │ ├── maxpooling.rs │ │ ├── mod.rs │ │ ├── relu.rs │ │ ├── sigmoid.rs │ │ └── softmax.rs │ └── mod.rs ├── lib.rs ├── parser │ └── mod.rs ├── runtime │ ├── mod.rs │ ├── pnnx │ │ ├── graph.rs │ │ ├── mod.rs │ │ ├── parameter.rs │ │ └── store_zip.rs │ ├── runtime_attribute.rs │ ├── runtime_datetype.rs │ ├── runtime_graph.rs │ ├── runtime_operand.rs │ ├── runtime_operator.rs │ └── runtime_parameter.rs └── status_code.rs └── tests ├── test_runtime_test_param.rs └── test_tensor.rs /.gitignore: -------------------------------------------------------------------------------- 1 | /target 2 | /笔记 3 | .vscode/ 4 | -------------------------------------------------------------------------------- /Cargo.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/Cargo.lock -------------------------------------------------------------------------------- /Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/Cargo.toml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/README.md -------------------------------------------------------------------------------- /examples/test_resnet.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/examples/test_resnet.rs -------------------------------------------------------------------------------- /model_file/car.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/model_file/car.jpg -------------------------------------------------------------------------------- /model_file/resnet18_batch1.pnnx.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/model_file/resnet18_batch1.pnnx.bin -------------------------------------------------------------------------------- /model_file/resnet18_batch1.pnnx.param: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/model_file/resnet18_batch1.pnnx.param -------------------------------------------------------------------------------- /model_file/simple_ops.pnnx.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/model_file/simple_ops.pnnx.bin -------------------------------------------------------------------------------- /model_file/simple_ops.pnnx.param: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/model_file/simple_ops.pnnx.param -------------------------------------------------------------------------------- /model_file/simple_ops2.pnnx.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/model_file/simple_ops2.pnnx.bin -------------------------------------------------------------------------------- /model_file/simple_ops2.pnnx.param: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/model_file/simple_ops2.pnnx.param -------------------------------------------------------------------------------- /model_file/test_linear.pnnx.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/model_file/test_linear.pnnx.bin -------------------------------------------------------------------------------- /model_file/test_linear.pnnx.param: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/model_file/test_linear.pnnx.param -------------------------------------------------------------------------------- /model_file/yolov5s.pnnx.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/model_file/yolov5s.pnnx.bin -------------------------------------------------------------------------------- /model_file/yolov5s.pnnx.param: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/model_file/yolov5s.pnnx.param -------------------------------------------------------------------------------- /src/data/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/data/mod.rs -------------------------------------------------------------------------------- /src/data/tensor.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/data/tensor.rs -------------------------------------------------------------------------------- /src/data/torch_utils.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/data/torch_utils.rs -------------------------------------------------------------------------------- /src/layer/abstract_layer/layer.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/layer/abstract_layer/layer.rs -------------------------------------------------------------------------------- /src/layer/abstract_layer/layer_factory.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/layer/abstract_layer/layer_factory.rs -------------------------------------------------------------------------------- /src/layer/abstract_layer/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/layer/abstract_layer/mod.rs -------------------------------------------------------------------------------- /src/layer/details/adaptive_avgpooling.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/layer/details/adaptive_avgpooling.rs -------------------------------------------------------------------------------- /src/layer/details/convolution.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/layer/details/convolution.rs -------------------------------------------------------------------------------- /src/layer/details/expression.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/layer/details/expression.rs -------------------------------------------------------------------------------- /src/layer/details/flatten.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/layer/details/flatten.rs -------------------------------------------------------------------------------- /src/layer/details/linear.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/layer/details/linear.rs -------------------------------------------------------------------------------- /src/layer/details/maxpooling.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/layer/details/maxpooling.rs -------------------------------------------------------------------------------- /src/layer/details/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/layer/details/mod.rs -------------------------------------------------------------------------------- /src/layer/details/relu.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/layer/details/relu.rs -------------------------------------------------------------------------------- /src/layer/details/sigmoid.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/layer/details/sigmoid.rs -------------------------------------------------------------------------------- /src/layer/details/softmax.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/layer/details/softmax.rs -------------------------------------------------------------------------------- /src/layer/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/layer/mod.rs -------------------------------------------------------------------------------- /src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/lib.rs -------------------------------------------------------------------------------- /src/parser/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/parser/mod.rs -------------------------------------------------------------------------------- /src/runtime/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/runtime/mod.rs -------------------------------------------------------------------------------- /src/runtime/pnnx/graph.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/runtime/pnnx/graph.rs -------------------------------------------------------------------------------- /src/runtime/pnnx/mod.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/runtime/pnnx/mod.rs -------------------------------------------------------------------------------- /src/runtime/pnnx/parameter.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/runtime/pnnx/parameter.rs -------------------------------------------------------------------------------- /src/runtime/pnnx/store_zip.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/runtime/pnnx/store_zip.rs -------------------------------------------------------------------------------- /src/runtime/runtime_attribute.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/runtime/runtime_attribute.rs -------------------------------------------------------------------------------- /src/runtime/runtime_datetype.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/runtime/runtime_datetype.rs -------------------------------------------------------------------------------- /src/runtime/runtime_graph.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/runtime/runtime_graph.rs -------------------------------------------------------------------------------- /src/runtime/runtime_operand.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/runtime/runtime_operand.rs -------------------------------------------------------------------------------- /src/runtime/runtime_operator.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/runtime/runtime_operator.rs -------------------------------------------------------------------------------- /src/runtime/runtime_parameter.rs: -------------------------------------------------------------------------------- 1 | 2 | 3 | pub struct RuntimeParameter{ 4 | 5 | 6 | } -------------------------------------------------------------------------------- /src/status_code.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/src/status_code.rs -------------------------------------------------------------------------------- /tests/test_runtime_test_param.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/tests/test_runtime_test_param.rs -------------------------------------------------------------------------------- /tests/test_tensor.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SongQiPing/KuiperInfer_rs/HEAD/tests/test_tensor.rs --------------------------------------------------------------------------------