summaryrefslogtreecommitdiff
path: root/candle-core/src/backend.rs
diff options
context:
space:
mode:
Diffstat (limited to 'candle-core/src/backend.rs')
-rw-r--r--candle-core/src/backend.rs10
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,