From c04510590643b14afcf358063da4c7e7897fd7ed Mon Sep 17 00:00:00 2001
From: Corey Lowman <clowman1993@gmail.com>
Date: Sun, 2 Apr 2023 20:42:16 +0000
Subject: [PATCH] Removing blanket impls for all types

---
 src/tensor_ops/conv2d/mod.rs      | 2 +-
 src/tensor_ops/convtrans2d/mod.rs | 2 +-
 src/tensor_ops/pool2d/mod.rs      | 2 +-
 src/tensor_ops/upscale2d/mod.rs   | 2 +-
 tests/test_resnet18_f32.rs        | 2 +-
 5 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/tensor_ops/conv2d/mod.rs b/src/tensor_ops/conv2d/mod.rs
index 0b2f8374f..239aae93b 100644
--- 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
index acc94b110..3352a8f7b 100644
--- 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
index b462d8032..446c27a49 100644
--- 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
index 433a83944..bfd2057fc 100644
--- 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
index 3f55583c8..2243842e8 100644
--- 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::*;