coreylowman · coreylowman · Apr 2, 2023 · Apr 2, 2023
diff --git a/src/tensor_ops/conv2d/mod.rs b/src/tensor_ops/conv2d/mod.rs
@@ -130,7 +130,7 @@ pub trait TryConv2D<F> {
     }
 }
 
-impl<T, F> TryConv2D<F> for T {}
+impl<S: Shape, E: Dtype, D: DeviceStorage, T, F> TryConv2D<F> for Tensor<S, E, D, T> {}
 
 impl<
         const C: usize,

diff --git a/src/tensor_ops/convtrans2d/mod.rs b/src/tensor_ops/convtrans2d/mod.rs
@@ -128,7 +128,7 @@ pub trait TryConvTrans2D<F> {
     }
 }
 
-impl<T, F> TryConvTrans2D<F> for T {}
+impl<S: Shape, E: Dtype, D: DeviceStorage, T, F> TryConvTrans2D<F> for Tensor<S, E, D, T> {}
 
 impl<
         const C: usize,

diff --git a/src/tensor_ops/pool2d/mod.rs b/src/tensor_ops/pool2d/mod.rs
@@ -81,7 +81,7 @@ macro_rules! pool2d {
                 self.try_pool2d()
             }
         }
-        impl<T> $TryTrait for T {}
+        impl<S: Shape, E: Dtype, D: DeviceStorage, T> $TryTrait for Tensor<S, E, D, T> {}
 
         impl<
                 C: Dim,

diff --git a/src/tensor_ops/upscale2d/mod.rs b/src/tensor_ops/upscale2d/mod.rs
@@ -123,7 +123,7 @@ pub trait TryUpscale2D {
         Upscale2DWithMethod::try_upscale2d_like(self, height, width)
     }
 }
-impl<T> TryUpscale2D for T {}
+impl<S: Shape, E: Dtype, D: DeviceStorage, T> TryUpscale2D for Tensor<S, E, D, T> {}
 
 impl<
         C: Dim,

diff --git a/tests/test_resnet18_f32.rs b/tests/test_resnet18_f32.rs
@@ -1,6 +1,6 @@
 #![cfg_attr(feature = "nightly", feature(generic_const_exprs))]
 
-#[cfg(feature = "nightly")]
+#[cfg(all(feature = "test-integrations", feature = "nightly"))]
 mod model {
     use dfdx::prelude::*;
-Original file line number
+Diff line change
@@ Expand Up / @@ -130,7 +130,7 @@ pub trait TryConv2D<F> { @@
         }
     }
-    impl<T, F> TryConv2D<F> for T {}
+    impl<S: Shape, E: Dtype, D: DeviceStorage, T, F> TryConv2D<F> for Tensor<S, E, D, T> {}
     impl<
             const C: usize,
@@ Expand Down @@