diff options
Diffstat (limited to 'candle-core/src/op.rs')
| -rw-r--r-- | candle-core/src/op.rs | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/candle-core/src/op.rs b/candle-core/src/op.rs index d36aa301..de5094bd 100644 --- a/candle-core/src/op.rs +++ b/candle-core/src/op.rs @@ -66,6 +66,8 @@ pub(crate) enum Op { Reduce(Tensor, ReduceOp, Vec<usize>), Matmul(Tensor, Tensor), Embedding(Tensor, Tensor), + Gather(Tensor, Tensor, usize), + ScatterAdd(Tensor, Tensor, Tensor, usize), IndexSelect(Tensor, Tensor, usize), IndexAdd(Tensor, Tensor, Tensor, usize), WhereCond(Tensor, Tensor, Tensor), |