diff options
Diffstat (limited to 'candle-core/src/backend.rs')
-rw-r--r-- | candle-core/src/backend.rs | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/candle-core/src/backend.rs b/candle-core/src/backend.rs index 2b873e6e..8815c08d 100644 --- a/candle-core/src/backend.rs +++ b/candle-core/src/backend.rs @@ -40,6 +40,16 @@ pub trait BackendStorage: Sized { ) -> Result<Self>; fn embedding(&self, _: &Layout, _: &Self, _: &Layout) -> Result<Self>; + fn gather(&self, _: &Layout, _: &Self, _: &Layout, _: usize) -> Result<Self>; + fn scatter_add( + &self, + _: &Layout, + _: &Self, + _: &Layout, + _: &Self, + _: &Layout, + _: usize, + ) -> Result<Self>; fn index_select(&self, _: &Self, _: &Layout, _: &Layout, _: usize) -> Result<Self>; fn index_add( &self, |