diff options
Diffstat (limited to 'tensorflow/core/kernels/fill_functor.cc')
| -rw-r--r-- | tensorflow/core/kernels/fill_functor.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/fill_functor.cc b/tensorflow/core/kernels/fill_functor.cc index 08ec4baff3..0df8f9d3ed 100644 --- a/tensorflow/core/kernels/fill_functor.cc +++ b/tensorflow/core/kernels/fill_functor.cc @@ -62,6 +62,8 @@ void SetZeroFunctor<Eigen::SyclDevice, T>::operator()( #define DEFINE_SETZERO_SYCL(T) \ template struct SetZeroFunctor<Eigen::SyclDevice, T>; DEFINE_SETZERO_SYCL(float); +DEFINE_SETZERO_SYCL(bool); +DEFINE_SETZERO_SYCL(double); #undef DEFINE_SETZERO_SYCL #endif // TENSORFLOW_USE_SYCL |