diff options
Diffstat (limited to 'tensorflow/core/kernels/slice_op.cc')
-rw-r--r-- | tensorflow/core/kernels/slice_op.cc | 9 |
1 files changed, 3 insertions, 6 deletions
diff --git a/tensorflow/core/kernels/slice_op.cc b/tensorflow/core/kernels/slice_op.cc index 2a9ff40f8c..ee6f9a28cd 100644 --- a/tensorflow/core/kernels/slice_op.cc +++ b/tensorflow/core/kernels/slice_op.cc @@ -328,8 +328,9 @@ namespace functor { DECLARE_SYCL_SPEC(T, 6); \ DECLARE_SYCL_SPEC(T, 7); -TF_CALL_GPU_NUMBER_TYPES(DECLARE_FOR_N); +TF_CALL_GPU_NUMBER_TYPES_NO_HALF(DECLARE_FOR_N); DECLARE_FOR_N(int32); +DECLARE_FOR_N(bool); #undef DECLARE_FOR_N #undef DECLARE_SYCL_SPEC @@ -344,11 +345,8 @@ DECLARE_FOR_N(int32); .TypeConstraint<int32>("Index"), \ SliceOp<SYCLDevice, type>) -TF_CALL_GPU_NUMBER_TYPES(REGISTER_SYCL); +TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER_SYCL); -// A special GPU kernel for int32. -// TODO(b/25387198): Also enable int32 in device memory. This kernel -// registration requires all int32 inputs and outputs to be in host memory. REGISTER_KERNEL_BUILDER(Name("Slice") .Device(DEVICE_SYCL) .TypeConstraint<int32>("T") @@ -358,7 +356,6 @@ REGISTER_KERNEL_BUILDER(Name("Slice") .HostMemory("size") .HostMemory("output"), SliceOp<CPUDevice, int32>); - #undef REGISTER_SYCL #endif // TENSORFLOW_USE_SYCL |