summaryrefslogtreecommitdiff
path: root/candle-onnx
Commit message (Collapse)AuthorAgeFilesLines
...
* [ONNX] Support a couple more ops. (#1284)Laurent Mazare2023-11-062-113/+109
| | | | | | | | | | | | | * Support the shape op in ONNX. * Share the axis normalization bits. * Add some limited support for gather. * Unsqueeze. * Comparison with broadcasting. * Add Not + handle i32.
* Put the onnx example behind a feature flag. (#1276)Laurent Mazare2023-11-062-9/+31
| | | | | | | * Put the onnx example behind a feature flag. * Exclude the onnx bits from the workspace. * README tweaks.
* Add more models to the onnx example. (#1273)Laurent Mazare2023-11-051-15/+152
| | | | | | | | | | | | | | | * Add more models to the onnx example. * Input validation. * Input validation. * Bugfix. * Implement clip. * BatchNorm support. * Get the efficientnet onnx to work.
* [ONNX] Do not generate values for constants. (#1272)Laurent Mazare2023-11-052-37/+38
| | | | | * Do not generate values for constants. * Add an onnx based example using squeezenet.
* Better tensor initialization in ONNX. (#1270)Laurent Mazare2023-11-041-10/+103
| | | | | | | | | * Better tensor initialization in ONNX. * MaxPool support. * Add AvgPool. * Get the squeezenet example to work.
* Refactor the onnx attribute getters. (#1268)Laurent Mazare2023-11-041-73/+218
| | | | | | | | | * Refactor the onnx attribute getters. * Add get-attr-opt. * Add support for convolutions. * Add support for convolutions.
* Support more ONNX ops. (#1267)Laurent Mazare2023-11-041-0/+49
| | | | | * Add LogSoftmax. * Support for Transpose.
* Improve the ONNX basic example + bugfixes (#1266)Laurent Mazare2023-11-043-52/+190
| | | | | | | | | | | | | * Generate some zeros tensor in the onnx simple-eval example. * Fix the casting operation. * Support more ops. * Handle reshape. * Concat. * Softmax.
* ONNX casting support. (#1265)Laurent Mazare2023-11-041-10/+94
| | | | | | | * ONNX casting support. * Handle tensor constants. * Bugfix the binary ops.
* Add some preliminary ONNX support (#1260)Laurent Mazare2023-11-046-0/+1015
* 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.