From bc00d41b4f96043748374ec58912c9ee90cbb601 Mon Sep 17 00:00:00 2001 From: Yong Tang Date: Sat, 23 Jun 2018 17:06:52 +0000 Subject: Register float16 for ImageProjectiveTransform ops in image_ops.cc Signed-off-by: Yong Tang --- tensorflow/contrib/image/ops/image_ops.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'tensorflow/contrib/image') diff --git a/tensorflow/contrib/image/ops/image_ops.cc b/tensorflow/contrib/image/ops/image_ops.cc index ebdcaea7ab..e59f1bf844 100644 --- a/tensorflow/contrib/image/ops/image_ops.cc +++ b/tensorflow/contrib/image/ops/image_ops.cc @@ -29,7 +29,7 @@ using shape_inference::ShapeHandle; REGISTER_OP("ImageProjectiveTransform") .Input("images: dtype") .Input("transforms: float32") - .Attr("dtype: {uint8, int32, int64, float32, float64}") + .Attr("dtype: {uint8, int32, int64, float16, float32, float64}") .Attr("interpolation: string") .Output("transformed_images: dtype") .SetShapeFn([](InferenceContext* c) { -- cgit v1.2.3