diff options
Diffstat (limited to 'tensorflow/core/kernels/reduction_ops_gpu.cu.cc')
| -rw-r--r-- | tensorflow/core/kernels/reduction_ops_gpu.cu.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/reduction_ops_gpu.cu.cc b/tensorflow/core/kernels/reduction_ops_gpu.cu.cc index a740bf1df3..d5d9f44da2 100644 --- a/tensorflow/core/kernels/reduction_ops_gpu.cu.cc +++ b/tensorflow/core/kernels/reduction_ops_gpu.cu.cc @@ -84,6 +84,7 @@ struct ReduceFunctor<GPUDevice, Eigen::internal::MeanReducer<T> > { DEFINE_FOR_TYPE_AND_R(T, Eigen::internal::ProdReducer<T>) DEFINE_FOR_ALL_REDUCERS(float); +DEFINE_FOR_ALL_REDUCERS(double); #undef DEFINE_FOR_ALL_REDUCERS DEFINE_FOR_TYPE_AND_R(complex64, Eigen::internal::SumReducer<complex64>); |