Update no-std

coreylowman · Oct 25, 2023 · aef72b0 · aef72b0
1 parent a4db3d5
commit aef72b0
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/dfdx-core/src/nn_traits/mod.rs b/dfdx-core/src/nn_traits/mod.rs
@@ -1,6 +1,8 @@
 mod tuples;
 mod vecs;
 
+use std::vec::Vec;
+
 use crate::prelude::{Device, Dtype, Gradients, Shape, Tensor, UniqueId};
 
 /// Mutable & Immutable forward of `Input` that produces [Module::Output].

diff --git a/dfdx-core/src/nn_traits/tuples.rs b/dfdx-core/src/nn_traits/tuples.rs
@@ -1,5 +1,7 @@
 use crate::{dtypes::Dtype, tensor::UniqueId, tensor_ops::Device};
 
+use std::vec::Vec;
+
 macro_rules! tuple_impls {
     ([$($name:ident),+] [$($idx:tt),+], $last:ident, [$($rev_tail:ident),*]) => {
 

diff --git a/dfdx-core/src/nn_traits/vecs.rs b/dfdx-core/src/nn_traits/vecs.rs
@@ -1,5 +1,7 @@
 use crate::{dtypes::Dtype, tensor::UniqueId, tensor_ops::Device};
 
+use std::vec::Vec;
+
 impl<E: Dtype, D: Device<E>, T: crate::nn_traits::BuildOnDevice<E, D>>
     crate::nn_traits::BuildOnDevice<E, D> for Vec<T>
 {