message: test for float types that GPU supports
diff: @@ -45,13 +45,14 @@ class ExtractVolumePatches(test.TestCase):
  ksizes = [1] + ksizes + [1]
  strides = [1] + strides + [1]
  + for dtype in [np.float16, np.float32, np.float64]:
  out_tensor = array_ops.extract_volume_patches(
  - constant_op.constant(image),
  + constant_op.constant(image.astype(dtype)),
  ksizes=ksizes,
  strides=strides,
  padding=padding,
  name="im2col_3d")
  - self.assertAllClose(patches, self.evaluate(out_tensor))
  + self.assertAllClose(patches.astype(dtype), self.evaluate(out_tensor))
  # pylint: disable=bad-whitespace
  def testKsize1x1x1Stride1x1x1(self):
  