eec11ce2ce
onnx: implement Size op ( #2316 )
2024-07-07 19:56:36 +02:00
2b10aaa05d
implement Slice op ( #2260 )
2024-06-12 07:15:32 +01:00
b9fac7ec00
implement if, and pad reflect mode ( #2251 )
...
* implement if, and pad reflect mode
The intent of this change is to allow eval of the current silero_vad.onnx (v4).
This onnx file uses 'If' and 'Pad' nodes, which had not been supported
by simple_eval until now
* Cleanup (fmt, clippy, minor test tweaks).
---------
Co-authored-by: Laurent <laurent.mazare@gmail.com >
2024-06-06 22:36:23 +02:00
d39462856b
Apply rustfmt. ( #2247 )
2024-06-04 22:54:09 +02:00
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
03344d3c19
ONNX: Add Floor and Ceil ( #2235 )
2024-06-02 21:45:20 +02:00
7ff921c538
Add RandomNormal ONNX operator ( #2200 )
2024-05-21 21:47:32 +02:00
7211009179
Fix for rustfmt. ( #2117 )
2024-04-23 19:09:33 +02:00
6fadaf2eff
candle-onnx: add operators RandomUniform and Exp ( #2116 )
...
* Add basic RandomUniform implementation
* Use is_some to check if seed is present
* Added Exp operator implementation
---------
Co-authored-by: Mateusz Okulus <mmokulus@gmail.com >
2024-04-23 19:02:19 +02:00
9215e9ce8c
Add missing onnx operations ( #2096 )
...
* Add missing onnx operations
* Add tests and fix errors
* Run rustfmt
2024-04-20 18:44:22 +02:00
e6d412b156
Add ReduceMean onnx operation ( #2049 )
...
* Add ReduceMean onnx operation
* Format code with rustfmt
2024-04-13 11:00:25 +02:00
798e0335cd
Handle more tensor shapes in onnx "Gather" operation ( #2026 )
...
* Handle more tensor shapes in onnx "Gather" operation
* Add more tests
* Add comment
* Fix typo
2024-04-08 14:06:14 +02:00
eeb7e2b683
Apply rustfmt to the newly added tests. ( #1749 )
2024-02-23 06:48:28 +01:00
11ea7aac4d
tests ( #1724 )
2024-02-23 06:35:46 +01:00
5cdd84e0f6
onnx: add the Flatten operator. ( #1638 )
...
* onnx: add the Flatten operator.
* onnx flatten: merge axis condition
---------
Co-authored-by: 王泽龙 <wangzelong@shenqishen.com >
2024-02-03 16:28:47 +01:00
8d8f48c60c
feat: add test for individual onnx ops ( #1332 )
...
* feat: add test for individual onnx ops
* fix: prefer consts when possible
* feat: add move op tests
2023-11-19 08:17:09 +01:00
