diff options
| author | MilkFather <31627231+MilkFather@users.noreply.github.com> | 2024-04-29 17:04:43 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-04-29 11:04:43 +0200 |
| commit | 3bbb88fcb463a6bdbb0e71c7b2d211dd02681493 (patch) | |
| tree | 6077a6f41f7b1ed97b6f44d5e8305126c0d5f5a5 /candle-examples/examples/quantized-phi/main.rs | |
| parent | ed7b99f525ab898aa677fe1f4446e345ac74f4ec (diff) | |
| download | candle-3bbb88fcb463a6bdbb0e71c7b2d211dd02681493.tar.gz candle-3bbb88fcb463a6bdbb0e71c7b2d211dd02681493.tar.bz2 candle-3bbb88fcb463a6bdbb0e71c7b2d211dd02681493.zip | |
Fix sigmoid gradient calculation and move sigmoid into a specialized op (#2114)
* add sigmoid op
* small fix
* add as a method on `Tensor`
* implement gradient calculation for sigmoid
* add sigmoid tests
* we should have a specialized op for this
* fix clippy
* fix clippy 2
* Revert all previous commits in favor of a `CustomOp` based solution
* use `CustomOp1` implementation
* fix rustfmt
* experimental add metal impl
* add cuda kernel impl
* fix fmt
* Add a test + reduce some cuda duplication.
---------
Co-authored-by: laurent <laurent.mazare@gmail.com>
Diffstat (limited to 'candle-examples/examples/quantized-phi/main.rs')
0 files changed, 0 insertions, 0 deletions