huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 11:56:45 +00:00
Code Issues Packages Projects Releases Wiki Activity
1,489 Commits 74 Branches 16 Tags
508f811b93035f076e18778fe08106f15abfa8a7
Commit Graph

10 Commits

Author SHA1 Message Date
Laurent Mazare
a773a4b22b [ONNX] Support a couple more ops. (#1284) 
* 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.
 2023-11-06 22:44:58 +01:00
Laurent Mazare
2a45bcf943 Put the onnx example behind a feature flag. (#1276) 
* Put the onnx example behind a feature flag.

* Exclude the onnx bits from the workspace.

* README tweaks.
 2023-11-06 07:45:07 +01:00
Laurent Mazare
f365a075e5 Add more models to the onnx example. (#1273) 
* Add more models to the onnx example.

* Input validation.

* Input validation.

* Bugfix.

* Implement clip.

* BatchNorm support.

* Get the efficientnet onnx to work.
 2023-11-05 16:57:26 +01:00
Laurent Mazare
928a9d906e [ONNX] Do not generate values for constants. (#1272) 
* Do not generate values for constants.

* Add an onnx based example using squeezenet.
 2023-11-05 11:23:14 +01:00
Laurent Mazare
39ad840a90 Better tensor initialization in ONNX. (#1270) 
* Better tensor initialization in ONNX.

* MaxPool support.

* Add AvgPool.

* Get the squeezenet example to work.
 2023-11-04 22:17:45 +01:00
Laurent Mazare
b5e4f84bed Refactor the onnx attribute getters. (#1268) 
* Refactor the onnx attribute getters.

* Add get-attr-opt.

* Add support for convolutions.

* Add support for convolutions.
 2023-11-04 21:31:48 +01:00
Laurent Mazare
dc68c130e4 Support more ONNX ops. (#1267) 
* Add LogSoftmax.

* Support for Transpose.
 2023-11-04 15:10:14 +01:00
Laurent Mazare
bc9a1bf239 Improve the ONNX basic example + bugfixes (#1266) 
* Generate some zeros tensor in the onnx simple-eval example.

* Fix the casting operation.

* Support more ops.

* Handle reshape.

* Concat.

* Softmax.
 2023-11-04 10:02:47 +01:00
Laurent Mazare
f7c957d64f ONNX casting support. (#1265) 
* ONNX casting support.

* Handle tensor constants.

* Bugfix the binary ops.
 2023-11-04 08:34:24 +01:00
Laurent Mazare
8cbb9d0e6c Add some preliminary ONNX support (#1260) 
* 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.
 2023-11-04 06:36:05 +01:00