diff options
Diffstat (limited to 'tensorflow/core/kernels/relu_op_gpu.cu.cc')
-rw-r--r-- | tensorflow/core/kernels/relu_op_gpu.cu.cc | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/relu_op_gpu.cu.cc b/tensorflow/core/kernels/relu_op_gpu.cu.cc index 30c4a289f7..ec09d8dfea 100644 --- a/tensorflow/core/kernels/relu_op_gpu.cu.cc +++ b/tensorflow/core/kernels/relu_op_gpu.cu.cc @@ -35,7 +35,9 @@ typedef Eigen::GpuDevice GPUDevice; template struct functor::Relu6<GPUDevice, T>; \ template struct functor::Relu6Grad<GPUDevice, T>; \ template struct functor::Elu<GPUDevice, T>; \ - template struct functor::EluGrad<GPUDevice, T>; + template struct functor::EluGrad<GPUDevice, T>; \ + template struct functor::Selu<GPUDevice, T>; \ + template struct functor::SeluGrad<GPUDevice, T>; TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_KERNELS); |