ENH: remove numeric type check

remicres · Mar 13, 2021 · fca60ed · fca60ed
1 parent 0c14c8d
commit fca60ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/code/ops.py b/code/ops.py
@@ -147,7 +147,7 @@ def _downscale2d(x, factor=2, gain=1):
     if gain != 1:
         x *= gain
 
-    if factor == 2 and x.dtype == tf.float32:
+    if factor == 2:
         return _blur2d(x, stride=factor)
 
     if factor == 1: