From 96344140077298606c256eaf32b7e86a788d04bc Mon Sep 17 00:00:00 2001
From: Luke Iwanski <luke@codeplay.com>
Date: Thu, 1 Jun 2017 03:30:28 +0100
Subject: [PATCH] [OpenCL] Extends softmax op to cover double (#10323)

---
 tensorflow/core/kernels/softmax_op.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tensorflow/core/kernels/softmax_op.cc b/tensorflow/core/kernels/softmax_op.cc
index de11de32f12..8345a98a0d3 100644
--- a/tensorflow/core/kernels/softmax_op.cc
+++ b/tensorflow/core/kernels/softmax_op.cc
@@ -90,6 +90,8 @@ REGISTER_KERNEL_BUILDER(
 REGISTER_KERNEL_BUILDER(
     Name("Softmax").Device(DEVICE_SYCL).TypeConstraint<float>("T"),
     SoftmaxOp<SYCLDevice, float>);
+REGISTER_KERNEL_BUILDER(
+    Name("Softmax").Device(DEVICE_SYCL).TypeConstraint<double>("T"),
+    SoftmaxOp<SYCLDevice, double>);
 #endif // TENSORFLOW_USE_SYCL
-
 }  // namespace tensorflow