diff options
Diffstat (limited to 'tensorflow/core/kernels/reshape_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/reshape_op.cc | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/tensorflow/core/kernels/reshape_op.cc b/tensorflow/core/kernels/reshape_op.cc index 6589a54624..04454b76c1 100644 --- a/tensorflow/core/kernels/reshape_op.cc +++ b/tensorflow/core/kernels/reshape_op.cc @@ -42,8 +42,12 @@ TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_GPU_KERNEL); .TypeConstraint<type>("T") \ .TypeConstraint<int32>("Tshape"), \ ReshapeOp); -TF_CALL_NUMBER_TYPES_NO_INT32(REGISTER_SYCL_KERNEL); -#undef REGISTER_SYCL_KERNEL +REGISTER_SYCL_KERNEL(float) +REGISTER_SYCL_KERNEL(double) +REGISTER_SYCL_KERNEL(uint8) +REGISTER_SYCL_KERNEL(int8) +REGISTER_SYCL_KERNEL(int64) +REGISTER_SYCL_KERNEL(uint16) REGISTER_KERNEL_BUILDER(Name("Reshape") .Device(DEVICE_SYCL) @@ -53,6 +57,7 @@ REGISTER_KERNEL_BUILDER(Name("Reshape") .TypeConstraint<int32>("T") .TypeConstraint<int32>("Tshape"), ReshapeOp); +#undef REGISTER_SYCL_KERNEL #endif // TENSORFLOW_USE_SYCL #if GOOGLE_CUDA |