diff options
author | 2018-06-23 17:03:38 +0000 | |
---|---|---|
committer | 2018-06-23 17:03:38 +0000 | |
commit | d2e3711f2931dd26c8716f5a229e2cb14781afde (patch) | |
tree | 90cd077dfd89a3a4c10a49824e84f0e1279a58f6 /tensorflow/contrib/image | |
parent | c1d223de41838e9d387a48137c76ea39d3b38f3f (diff) |
Add float16 support for tf.contrib.image.transform
This fix tries to address the issue raised in 20243 where
there were no float16 support for tf.contrib.image.transform.
This fix adds the float16 support and enables related tests.
This fix fixes 20243.
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
Diffstat (limited to 'tensorflow/contrib/image')
-rw-r--r-- | tensorflow/contrib/image/python/ops/image_ops.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/image/python/ops/image_ops.py b/tensorflow/contrib/image/python/ops/image_ops.py index cd984c8054..ab739799e7 100644 --- a/tensorflow/contrib/image/python/ops/image_ops.py +++ b/tensorflow/contrib/image/python/ops/image_ops.py @@ -33,7 +33,7 @@ _image_ops_so = loader.load_op_library( resource_loader.get_path_to_datafile("_image_ops.so")) _IMAGE_DTYPES = set( - [dtypes.uint8, dtypes.int32, dtypes.int64, dtypes.float32, dtypes.float64]) + [dtypes.uint8, dtypes.int32, dtypes.int64, dtypes.float16, dtypes.float32, dtypes.float64]) ops.RegisterShape("ImageConnectedComponents")(common_shapes.call_cpp_shape_fn) ops.RegisterShape("ImageProjectiveTransform")(common_shapes.call_cpp_shape_fn) |