MilkFather 3bbb88fcb4 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>

2024-04-29 11:04:43 +02:00

..

cudnn.rs

Backend refactoring. (#1966)

2024-03-29 23:02:11 +01:00

device.rs

Add StorageRef. (#2113)

2024-04-23 13:23:27 +02:00

error.rs

Split the cuda error file. (#2003)

2024-04-04 08:27:23 +02:00

mod.rs

Fix sigmoid gradient calculation and move sigmoid into a specialized op (#2114)

2024-04-29 11:04:43 +02:00

utils.rs

Backend refactoring. (#1966)

2024-03-29 23:02:11 +01:00