B1rtek cb180eb23a ONNX: add ArgMin, ArgMax and LeakyRelu (#2246) ...

* Add basic RandomUniform implementation

* Use is_some to check if seed is present

* Added Exp operator implementation

* Added ArgMin operator implementation

* Added tests for ArgMin

* ArgMin now returns a tensor with i64

* Added tests from pytorch examples

* Added ArgMax operator implementation

* Added tests for ArgMax

* Added LeakyRelu implementation

* Added a test for LeakyRelu

* Typo fix

* Fix a weird automatic RustRover change

---------

Co-authored-by: Mateusz Okulus <mmokulus@gmail.com>

2024-06-04 22:49:02 +02:00

..

ops.rs

ONNX: add ArgMin, ArgMax and LeakyRelu (#2246)

2024-06-04 22:49:02 +02:00