From bb9958ab69a38cbe57d119947b635a257fa6b802 Mon Sep 17 00:00:00 2001
From: James Qin <jamesqin@google.com>
Date: Mon, 17 Sep 2018 19:56:41 -0700
Subject: Register fp16 reduce_max on GPU

PiperOrigin-RevId: 213383647
---
 tensorflow/core/kernels/reduction_ops_max.cc | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'tensorflow/core/kernels/reduction_ops_max.cc')

diff --git a/tensorflow/core/kernels/reduction_ops_max.cc b/tensorflow/core/kernels/reduction_ops_max.cc
index 9cf953f4bf..8bfa44b2d0 100644
--- a/tensorflow/core/kernels/reduction_ops_max.cc
+++ b/tensorflow/core/kernels/reduction_ops_max.cc
@@ -50,6 +50,8 @@ TF_CALL_REAL_NUMBER_TYPES(REGISTER_CPU_KERNELS);
           .TypeConstraint<int64>("Tidx")                                       \
           .HostMemory("reduction_indices"),                                    \
       ReductionOp<GPUDevice, type, int64, Eigen::internal::MaxReducer<type>>);
+
+REGISTER_GPU_KERNELS(Eigen::half);
 REGISTER_GPU_KERNELS(float);
 REGISTER_GPU_KERNELS(double);
 REGISTER_GPU_KERNELS(int64);
-- 
cgit v1.2.3