diff options
Diffstat (limited to 'candle-core/src/backend.rs')
| -rw-r--r-- | candle-core/src/backend.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-core/src/backend.rs b/candle-core/src/backend.rs index c897510e..018279b3 100644 --- a/candle-core/src/backend.rs +++ b/candle-core/src/backend.rs @@ -16,7 +16,7 @@ pub(crate) trait BackendStorage: Sized { fn elu(&self, _: &Layout, _: f64) -> Result<Self>; - fn sum(&self, _: &Layout, _: &[usize]) -> Result<Self>; + fn reduce_op(&self, _: crate::op::ReduceOp, _: &Layout, _: &[usize]) -> Result<Self>; fn divide_by_sum_over_dim(&mut self, _: &Shape, _: usize) -> Result<()>; |