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.rs7
1 files changed, 3 insertions, 4 deletions
diff --git a/candle-core/src/backend.rs b/candle-core/src/backend.rs
index 307868dd..977dba69 100644
--- a/candle-core/src/backend.rs
+++ b/candle-core/src/backend.rs
@@ -1,4 +1,4 @@
-use crate::op::{CmpOp, ReduceOp};
+use crate::op::{BinaryOpT, CmpOp, ReduceOp, UnaryOpT};
use crate::{CpuStorage, DType, Layout, Result, Shape};
pub(crate) trait BackendStorage: Sized {
@@ -25,10 +25,9 @@ pub(crate) trait BackendStorage: Sized {
fn to_dtype(&self, _: &Layout, _: DType) -> Result<Self>;
- fn unary_impl<B: crate::op::UnaryOp>(&self, _: &Layout) -> Result<Self>;
+ fn unary_impl<B: UnaryOpT>(&self, _: &Layout) -> Result<Self>;
- fn binary_impl<B: crate::op::BinaryOp>(&self, _: &Self, _: &Layout, _: &Layout)
- -> Result<Self>;
+ fn binary_impl<B: BinaryOpT>(&self, _: &Self, _: &Layout, _: &Layout) -> Result<Self>;
fn where_cond(&self, _: &Layout, _: &Self, _: &Layout, _: &Self, _: &Layout) -> Result<Self>;