summaryrefslogtreecommitdiff
path: root/candle-onnx/src/lib.rs
Commit message (Collapse)AuthorAgeFilesLines
* PyO3: Add optional `candle.onnx` module (#1282)Lukas Kreussel2023-11-081-1/+1
| | | | | | | | | | | | | | | | | | | | | | | | | | | * Start onnx integration * Merge remote-tracking branch 'upstream/main' into feat/pyo3-onnx * Implement ONNXModel * `fmt` * add `onnx` flag to python ci * Pin `protoc` to `25.0` * Setup `protoc` in wheel builds * Build wheels with `onnx` * Install `protoc` in manylinux containers * `apt` -> `yum` * Download `protoc` via bash script * Back to `manylinux: auto` * Disable `onnx` builds for linux
* Improve the ONNX basic example + bugfixes (#1266)Laurent Mazare2023-11-041-1/+1
| | | | | | | | | | | | | * Generate some zeros tensor in the onnx simple-eval example. * Fix the casting operation. * Support more ops. * Handle reshape. * Concat. * Softmax.
* Add some preliminary ONNX support (#1260)Laurent Mazare2023-11-041-0/+14
* Add the onnx protos. * Move the reading bits. * Install protoc on the CI. * Install protoc on the cuda CI too. * Use clap for the onnx tool. * Tweak the CI protoc install. * Add some simple evalution function. * Add some binary operator support.