diff options
Diffstat (limited to 'tensorflow/core/kernels/scatter_nd_op_cpu_impl.h')
| -rw-r--r-- | tensorflow/core/kernels/scatter_nd_op_cpu_impl.h | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/tensorflow/core/kernels/scatter_nd_op_cpu_impl.h b/tensorflow/core/kernels/scatter_nd_op_cpu_impl.h index 7cfffa20c5..472f5a3547 100644 --- a/tensorflow/core/kernels/scatter_nd_op_cpu_impl.h +++ b/tensorflow/core/kernels/scatter_nd_op_cpu_impl.h @@ -161,15 +161,16 @@ struct ScatterNdFunctor<CPUDevice, T, Index, OP, IXDIM> { TF_CALL_ALL_TYPES(REGISTER_SCATTER_ND_UPDATE); REGISTER_SCATTER_ND_INDEX(string, scatter_nd_op::UpdateOp::ADD); -TF_CALL_NUMBER_TYPES(REGISTER_SCATTER_ND_MATH) - +TF_CALL_NUMBER_TYPES(REGISTER_SCATTER_ND_MATH); +TF_CALL_bool(REGISTER_SCATTER_ND_MATH); #undef REGISTER_SCATTER_ND_MATH #undef REGISTER_SCATTER_ND_UPDATE #undef REGISTER_SCATTER_ND_INDEX #undef REGISTER_SCATTER_ND_FULL -#ifdef TENSORFLOW_USE_SYCL // Implementation of update functor for SYCL. +#ifdef TENSORFLOW_USE_SYCL + template <typename T, typename Index, scatter_nd_op::UpdateOp OP, int IXDIM> struct ScatterNdFunctor<SYCLDevice, T, Index, OP, IXDIM> { Index operator()( |