diff options
Diffstat (limited to 'tensorflow/core/kernels/cwise_op_select.cc')
| -rw-r--r-- | tensorflow/core/kernels/cwise_op_select.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/cwise_op_select.cc b/tensorflow/core/kernels/cwise_op_select.cc index 709628da13..3dd9de8d89 100644 --- a/tensorflow/core/kernels/cwise_op_select.cc +++ b/tensorflow/core/kernels/cwise_op_select.cc @@ -181,7 +181,9 @@ REGISTER_SELECT_GPU(complex128); SelectOp<SYCLDevice, type>); REGISTER_SELECT_SYCL(float); +REGISTER_SELECT_SYCL(double); REGISTER_SELECT_SYCL(int32); +REGISTER_SELECT_SYCL(int64); #undef REGISTER_SELECT_SYCL #endif // TENSORFLOW_USE_SYCL |