diff options
Diffstat (limited to 'candle-core/src/backend.rs')
| -rw-r--r-- | candle-core/src/backend.rs | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/candle-core/src/backend.rs b/candle-core/src/backend.rs index 345db0e5..a8e5ac52 100644 --- a/candle-core/src/backend.rs +++ b/candle-core/src/backend.rs @@ -37,6 +37,17 @@ pub trait BackendStorage: Sized { _params: &crate::conv::ParamsConv1D, ) -> Result<Self>; + fn conv2d( + &self, + _l: &Layout, + _kernel: &Self, + _kernel_l: &Layout, + _params: &crate::conv::ParamsConv2D, + ) -> Result<Self>; + + fn avg_pool2d(&self, _: &Layout, _: (usize, usize), _: (usize, usize)) -> Result<Self>; + fn upsample_nearest2d(&self, _: &Layout, _: usize, _: usize) -> Result<Self>; + fn gather(&self, _: &Layout, _: &Self, _: &Layout, _: usize) -> Result<Self>; fn scatter_add( &self, |