summaryrefslogtreecommitdiff
path: root/candle-onnx
Commit message (Collapse)AuthorAgeFilesLines
* Bump the caret version to 0.8.2. (#2703)Laurent Mazare2025-01-071-3/+3
|
* Bump the crate version to 0.8.1. (#2662)Laurent Mazare2024-12-071-3/+3
|
* Onnx Support for Sign operation #2641 (#2642)Ionut Mihalcea2024-11-262-0/+47
| | | | | | | | | * Support for Sign operation #2641 * Apply rustfmt. --------- Co-authored-by: Laurent <laurent.mazare@gmail.com>
* Bump the crate version to 0.8.0. (#2612)Laurent Mazare2024-11-121-3/+3
|
* ONNX: GatherElements, Xor (#2568)Anubhab Bandyopadhyay2024-10-172-0/+582
|
* onnx: ReduceMin/Max Ops (#2563)Anubhab Bandyopadhyay2024-10-152-1/+1211
| | | | | | | | | | | | | | | | | | | | | | | * Stella_en_1.5B_v5 * Separated creation. This is a critical step for numerical accuracy and would be documented in the readme * EmbedDim would require clone and copy * WIP: example * Examples added * a litte more in README * WIP: ONNX Reduce-max ops * WIP: tests for ReduceMin * Reduce min/ max v18+ * Reformatting tests for better review readability * Error on empty set, backward compatibility (13 and below) with 'axes'
* Bump the crate version to 0.7.2. (#2517)Laurent Mazare2024-09-291-3/+3
|
* Clippy fixes for onnx + fix a broken test. (#2510)Laurent Mazare2024-09-262-281/+273
|
* Expand split ops (#2505)Steven Lovegrove2024-09-262-14/+528
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | * candle-onnx: Add Split and Expand operators, Fix Where Op Implemented based on https://github.com/onnx/onnx/blob/main/docs/Operators.md Test cases based on those examples. TODO: Should add the remaining Split examples as tests TODO: Add.test case that motivates Where fix * candle-onnx: Add ReduceSum operator Implemented based on https://github.com/onnx/onnx/blob/main/docs/Operators.md Test cases based on those examples. TODO: Should add the remaining ReduceSum examples as tests * candle-onnx: Add ReduceL2 operator Implemented based on https://github.com/onnx/onnx/blob/main/docs/Operators.md Test cases based on those examples. TODO: Should add the remaining ReduceSum examples as tests * candle-onnx: Fix Clip operator empty string as default arg issue Optional input args may be signified by an empty string. The length of the input array is not enough because non optional args may follow optional ones. I encountered this when trying to use the ONNX model found at https://huggingface.co/sentence-transformers/all-MiniLM-L6-v2 for example. The LSTM op has a utility which I factored to be more generally accessible, and I have used it in the ops I have recently created or debugged. I believe it is likely that this issue may also manifest in other ops, but I didn't want to change anything that I'm not testing. * fix formatting * fix small mistake made during refactor
* Move the candle version to 0.7.1. (#2495)Laurent Mazare2024-09-221-3/+3
|
* Bump the crate version. (#2491)Laurent Mazare2024-09-211-3/+3
|
* onnx: support negative index in Gather (#2440)shua2024-08-221-0/+12
| | | | | | index_select does not support negative indexing, but this change adds just enough workarounds in onnx to allow evaluating silero-vad models (which make use of negative indices).
* onnx: workaround pow with negative base (#2439)shua2024-08-221-2/+9
| | | | | | | | | | | | * onnx: workaround pow with negative base rather than fully defining pow in the cpu backend (as in #2318), this implements a much smaller change which is sufficient to evaluate silero-vad onnx models. Specifically, checking if pow is run with 2.0 exponent, and if so evaluate as simply `x*x` instead of the cpu backend of `e^(2.0 * ln(x))`. * PR: use Tensor::powf insead powf correctly handles a negative base.
* Bump the version to 0.6.1. (#2438)Laurent Mazare2024-08-221-3/+3
|
* onnx: implement LSTM op (#2268)shua2024-08-192-0/+705
| | | use candle-nn LSTM
* onnx: fix pad, unsqueeze (#2317)shua2024-07-232-7/+74
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | * onnx: fix pad, unsqueeze both implementations have off-by-one errors: - Pad 'reflect' cycle for eg `dim==3` is `[0,1,2,1]` which has length of 4 (or `dim*2 - 2`) not 5 (current code `dim*2 - 1`) - Unsqueeze(-1) for tensor with `dim==3` should be 3 (ie `dim+index+1`) not 2 (ie currently `dim+index`) in addition, Pad is incorrectly calculating the starting padding. If we want to pad out 2 elements to the start, and we have this cycle of indices of length 6, then we should skip 4 elements, but currently we skip 2. A more visual representation of what's going on is below: ``` pad_start: 2 data: [a,b,c,d] indices: [0, 1, 2, 3, 2, 1, 0, 1, 2, 3, 2, 1, 0, ..] // zigzag between 0..4 actual: skip [ c d| c b a b] expected: ~ skip ~ [ c b| a b c d] ``` The values between `[` and `|` are padding and the values between `|` and `]` in the example should match the original data being padded. * Fix clippy lints. --------- Co-authored-by: Laurent <laurent.mazare@gmail.com>
* onnx: implement Size op (#2316)shua2024-07-072-0/+52
|
* Adding Gemm and ArgMax operators to candle-onnx (#2231)drCathieSo.eth2024-06-281-0/+24
| | | | | | | | | | | | | * feat(gemm): implement Gemm operator in candle-onnx * feat(onnx): Add support for ArgMax operator in candle-onnx * Apply rustfmt. * Remove argmax as it was already present. --------- Co-authored-by: Laurent <laurent.mazare@gmail.com>
* implement Slice op (#2260)shua2024-06-122-0/+215
|
* implement if, and pad reflect mode (#2251)shua2024-06-062-54/+271
| | | | | | | | | | | | | * 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>
* Bump the crate version. (#2248)Laurent Mazare2024-06-051-3/+3
|
* Apply rustfmt. (#2247)Laurent Mazare2024-06-042-73/+64
|
* ONNX: add ArgMin, ArgMax and LeakyRelu (#2246)B1rtek2024-06-042-0/+523
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | * 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>
* ONNX: Add Floor and Ceil (#2235)mokulus2024-06-022-0/+162
|
* Add RandomNormal ONNX operator (#2200)mokulus2024-05-212-8/+158
|
* Bump the version number to 0.5.1. (#2155)Laurent Mazare2024-05-031-3/+3
| | | | | | | * Bump the version number to 0.5.1. * Fix clippy lints for 1.78. * More clippy fixes.
* Fix for rustfmt. (#2117)Laurent Mazare2024-04-231-5/+1
|
* candle-onnx: add operators RandomUniform and Exp (#2116)B1rtek2024-04-232-0/+241
| | | | | | | | | | | * 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>
* Add missing onnx operations (#2096)Gabriel2024-04-202-9/+736
| | | | | | | * Add missing onnx operations * Add tests and fix errors * Run rustfmt
* Add ReduceMean onnx operation (#2049)Gabriel2024-04-132-1/+201
| | | | | * Add ReduceMean onnx operation * Format code with rustfmt
* Handle more tensor shapes in onnx "Gather" operation (#2026)Gabriel2024-04-082-9/+152
| | | | | | | | | * Handle more tensor shapes in onnx "Gather" operation * Add more tests * Add comment * Fix typo
* Bumping the version number to 0.5.0. (#2009)Laurent Mazare2024-04-041-3/+3
|
* add identity op (#1976)Mauro Sciancalepore2024-04-011-0/+5
|
* Bump the crate versions to 0.4.2. (#1821)Laurent Mazare2024-03-081-3/+3
|
* Bump the version number to 0.4.1. (#1768)Laurent Mazare2024-02-271-3/+3
| | | | | * Fix the block size for some cuda kernels. * Bump the version number to 0.4.1.
* Apply rustfmt to the newly added tests. (#1749)Laurent Mazare2024-02-231-7/+7
|
* tests (#1724)Sacha Arbonel2024-02-231-10/+473
|
* Bump the crate version to 0.4.0. (#1658)Laurent Mazare2024-02-041-3/+3
|
* onnx: add the Flatten operator. (#1638)wanglong0012024-02-032-1/+139
| | | | | | | | | * onnx: add the Flatten operator. * onnx flatten: merge axis condition --------- Co-authored-by: 王泽龙 <wangzelong@shenqishen.com>
* Explicit version for packages that are not in the workspace. (#1642)Laurent Mazare2024-01-311-2/+2
|
* Add the pow operator. (#1583)Laurent Mazare2024-01-131-0/+6
| | | | | * Add the pow operator. * Support the pow operation in onnx.
* Unpin more of the workplace relative dependencies. (#1535)Laurent Mazare2024-01-071-2/+2
|
* Simplifying our internal cargo dependencies. (#1529)Nicolas Patry2024-01-071-1/+0
|
* Bump the crate version to 0.3.3. (#1490)Laurent Mazare2023-12-281-3/+3
|
* Bump the crate version to 0.3.2. (#1452)Laurent Mazare2023-12-171-3/+3
|
* feat: add test for individual onnx ops (#1332)drbh2023-11-191-0/+746
| | | | | | | * feat: add test for individual onnx ops * fix: prefer consts when possible * feat: add move op tests
* Support for CumSum in ONNX models. (#1340)Laurent Mazare2023-11-171-0/+19
|
* Update for 0.3.1. (#1324)Laurent Mazare2023-11-111-3/+3
|
* fix: negative axis (#1296)YangNianYi2023-11-081-29/+4
| | | | | | | | | * fix: negative axis * Use normalize_axis. --------- Co-authored-by: Laurent <laurent.mazare@gmail.com>
* PyO3: Add optional `candle.onnx` module (#1282)Lukas Kreussel2023-11-082-2/+2
| | | | | | | | | | | | | | | | | | | | | | | | | | | * Start onnx integration * Merge remote-tracking branch 'upstream/main' into feat/pyo3-onnx * Implement ONNXModel * `fmt` * add `onnx` flag to python ci * Pin `protoc` to `25.0` * Setup `protoc` in wheel builds * Build wheels with `onnx` * Install `protoc` in manylinux containers * `apt` -> `yum` * Download `protoc` via bash script * Back to `manylinux: auto` * Disable `onnx` builds for linux
generated by cgit v1.2.3 (git 2.39.1) at 2025-04-04 21:43:10 +0000