diff options
| author |  Yong Tang <yong.tang.github@outlook.com> | 2018-06-23 17:06:52 +0000 |
|---|---|---|
| committer | Yong Tang <yong.tang.github@outlook.com> | 2018-06-23 17:06:52 +0000 |
| commit | bc00d41b4f96043748374ec58912c9ee90cbb601 (patch) | |
| tree | b1a13c15d2279351fb53f5ea3b2baf1f3ac90b5d /tensorflow/contrib/image | |
| parent | fad970949ecc1294d255b6604e644f1bb338677a (diff) | |
Register float16 for ImageProjectiveTransform ops in image_ops.cc
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
Diffstat (limited to 'tensorflow/contrib/image')
| -rw-r--r-- | tensorflow/contrib/image/ops/image_ops.cc | 2 |
1 files changed, 1 insertions, 1 deletions
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) { |