diff options
| author | laurent <laurent.mazare@gmail.com> | 2023-07-21 20:12:48 +0100 |
|---|---|---|
| committer | laurent <laurent.mazare@gmail.com> | 2023-07-21 20:12:48 +0100 |
| commit | 27174a82aaef01f2457e80b0d5ed2a8d8c04dd28 (patch) | |
| tree | edbfa21e24a417f9533ef3cc375854eb0b322ab7 /candle-core/src/op.rs | |
| parent | 5cc843550d454f4e8143a915add7cb16db7d05cc (diff) | |
| download | candle-27174a82aaef01f2457e80b0d5ed2a8d8c04dd28.tar.gz candle-27174a82aaef01f2457e80b0d5ed2a8d8c04dd28.tar.bz2 candle-27174a82aaef01f2457e80b0d5ed2a8d8c04dd28.zip | |
Start adding index-add.
Diffstat (limited to 'candle-core/src/op.rs')
| -rw-r--r-- | candle-core/src/op.rs | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/candle-core/src/op.rs b/candle-core/src/op.rs index a33dd226..d36aa301 100644 --- a/candle-core/src/op.rs +++ b/candle-core/src/op.rs @@ -67,6 +67,7 @@ pub(crate) enum Op { Matmul(Tensor, Tensor), Embedding(Tensor, Tensor), IndexSelect(Tensor, Tensor, usize), + IndexAdd(Tensor, Tensor, Tensor, usize), WhereCond(Tensor, Tensor, Tensor), #[allow(dead_code)] |