diff options
Diffstat (limited to 'tensorflow/core/kernels/sequence_ops.cc')
| -rw-r--r-- | tensorflow/core/kernels/sequence_ops.cc | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/sequence_ops.cc b/tensorflow/core/kernels/sequence_ops.cc index e2577620d9..f8e6c22ab6 100644 --- a/tensorflow/core/kernels/sequence_ops.cc +++ b/tensorflow/core/kernels/sequence_ops.cc @@ -89,6 +89,12 @@ class RangeOp : public OpKernel { #define REGISTER_CPU_KERNEL(T) REGISTER_KERNEL(DEVICE_CPU, T) #define REGISTER_GPU_KERNEL(T) REGISTER_KERNEL(DEVICE_GPU, T) +#ifdef TENSORFLOW_USE_SYCL +#define REGISTER_SYCL_KERNEL(T) REGISTER_KERNEL(DEVICE_SYCL, T) +TF_CALL_float(REGISTER_SYCL_KERNEL); +TF_CALL_int32(REGISTER_SYCL_KERNEL); +TF_CALL_int64(REGISTER_SYCL_KERNEL); +#endif // TENSORFLOW_USE_SYCL TF_CALL_float(REGISTER_CPU_KERNEL); TF_CALL_double(REGISTER_CPU_KERNEL); |