make device capable of stacking and concatenating usize type, which a…

…re necessary for preparing tensors for embedding (#882) Co-authored-by: Corey Lowman <clowman1993@gmail.com>
coreylowman · Nov 6, 2023 · 4476b5e · 4476b5e
1 parent 53469e9
commit 4476b5e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/dfdx-core/src/tensor_ops/utilities/device.rs b/dfdx-core/src/tensor_ops/utilities/device.rs
@@ -15,8 +15,11 @@ pub trait Device<E: Dtype>:
 
     // appends
     + super::super::stack::StackKernel<E>
+    + super::super::stack::StackKernel<usize>
     + super::super::concat::ConcatKernel<E>
+    + super::super::concat::ConcatKernel<usize>
     + super::super::concat_along::ConcatAlongKernel<E>
+    + super::super::concat_along::ConcatAlongKernel<usize>
 
     // optimizers
     + super::super::adam::AdamKernel<E>