huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 11:08:52 +00:00
Code Issues Packages Projects Releases Wiki Activity
666 Commits 74 Branches 16 Tags
a0e47aba987dfd38b7fefc5639df58e1fb49dafc
Commit Graph

3 Commits

Author SHA1 Message Date
Laurent Mazare
5cc843550d Add binary and ternary custom ops. (#217) 2023-07-21 17:29:50 +01:00
Laurent Mazare
4a100875bf Use a macro to handle the dtype pattern matching. (#215) 2023-07-21 16:03:51 +01:00
Laurent Mazare
a6bcdfb269 Custom ops with a single argument (#214) 
* Add the CustomOp1 trait.

* Add an example of custom op.

* Polish the custom op example.

* Add some backward pass test for custom ops.
 2023-07-21 15:18:05 +01:00