|
|
2ced31b530
|
Added a test for LeakyRelu
|
2024-05-10 00:50:05 +02:00 |
|
|
|
4de76b89a2
|
Added tests for ArgMax
|
2024-05-09 20:45:53 +02:00 |
|
|
|
c4743aa570
|
Added tests from pytorch examples
|
2024-05-09 20:22:55 +02:00 |
|
|
|
13b88547f7
|
Added tests for ArgMin
|
2024-05-09 03:00:22 +02:00 |
|
|
|
a06b2ded28
|
Merge branch 'refs/heads/random' into operators-random-exp
# Conflicts:
# candle-onnx/tests/ops.rs
|
2024-04-23 17:36:33 +02:00 |
|
|
|
a867d652d3
|
Merge branch 'refs/heads/exp' into operators-random-exp
|
2024-04-23 17:33:05 +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 |
|
|
|
70388c27b6
|
Added Exp operator implementation
|
2024-04-19 22:48:05 +02:00 |
|
|
|
46073c5f73
|
Add basic RandomUniform implementation
|
2024-04-19 16:06:43 +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 |
|
