Don't assume convolution outputs to be exact integers (fixes issue #80).

PiperOrigin-RevId: 370087435
Change-Id: I9a7406c0a00d30aa6822824ade7b3087b50fb197

Author: Johannes Ballé

tensorflow_compression/python/ops/padding_ops_test.py

@@ -33,8 +33,8 @@ def test_same_padding_corr(self):
             tf.reshape(kernel, (1, -1, 1, 1)),
             padding="VALID", data_format="NHWC")
         outputs = np.squeeze(outputs.numpy())
-        pos_inp = np.squeeze(np.nonzero(inputs))
-        pos_out = np.squeeze(np.nonzero(outputs))
+        pos_inp = np.squeeze(np.nonzero(inputs > .5))
+        pos_out = np.squeeze(np.nonzero(outputs > .5))
         padding = padding_ops.same_padding_for_kernel(kshape, True)
         self.assertEqual(padding[0][0], pos_inp - pos_out)
 
@@ -52,8 +52,8 @@ def test_same_padding_conv(self):
             strides=(1, 1, 1, 1), padding="VALID", data_format="NHWC")
         outputs = outputs[:, :, (kshape[0] - 1):-(kshape[0] - 1), :]
         outputs = np.squeeze(outputs.numpy())
-        pos_inp = np.squeeze(np.nonzero(inputs))
-        pos_out = np.squeeze(np.nonzero(outputs))
+        pos_inp = np.squeeze(np.nonzero(inputs > .5))
+        pos_out = np.squeeze(np.nonzero(outputs > .5))
         padding = padding_ops.same_padding_for_kernel(kshape, False)
         self.assertEqual(padding[0][0], pos_inp - pos_out)