diff options
author | 2018-09-17 19:56:41 -0700 | |
---|---|---|
committer | 2018-09-17 20:01:03 -0700 | |
commit | bb9958ab69a38cbe57d119947b635a257fa6b802 (patch) | |
tree | 641e80043a5ed46d2c9c883b722af6c8e876fbf3 /tensorflow/core/kernels/reduction_ops_max.cc | |
parent | 1bd2804869355a7cd0cbfbe9e6aab7591b8a20de (diff) |
Register fp16 reduce_max on GPU
PiperOrigin-RevId: 213383647
Diffstat (limited to 'tensorflow/core/kernels/reduction_ops_max.cc')
-rw-r--r-- | tensorflow/core/kernels/reduction_ops_max.cc | 2 |
1 files changed, 2 insertions, 0 deletions
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); |