feat(ml): ML on Rockchip NPUs (#15241)

2025-03-18 00:04:08 +08:00
parent 1e184a70f1
commit 14c3b99c0f
43 changed files with 2417 additions and 4726 deletions
--- a/machine-learning/pyproject.toml
+++ b/machine-learning/pyproject.toml
@@ -51,6 +51,7 @@ cpu = ["onnxruntime>=1.15.0,<2"]
 cuda = ["onnxruntime-gpu>=1.17.0,<2"]
 openvino = ["onnxruntime-openvino>=1.17.1,<1.19.0"]
 armnn = ["onnxruntime>=1.15.0,<2"]
+rknn = ["onnxruntime>=1.15.0,<2", "rknn-toolkit-lite2>=2.3.0,<3"]

 [tool.uv]
 compile-bytecode = true