diff options
Diffstat (limited to 'tensorflow/core/kernels/fill_functor.h')
-rw-r--r-- | tensorflow/core/kernels/fill_functor.h | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/fill_functor.h b/tensorflow/core/kernels/fill_functor.h index 1032570fca..d6814cb477 100644 --- a/tensorflow/core/kernels/fill_functor.h +++ b/tensorflow/core/kernels/fill_functor.h @@ -45,6 +45,15 @@ struct SetZeroFunctor<Eigen::ThreadPoolDevice, T> { typename TTypes<T>::Flat out); }; +#ifdef TENSORFLOW_USE_SYCL +// Partial specialization of SetZeroFunctor<Device=Eigen::SyclDevice, T>. +template <typename T> +struct SetZeroFunctor<Eigen::SyclDevice, T> { + void operator()(const Eigen::SyclDevice& d, + typename TTypes<T>::Flat out); +}; +#endif // TENSORFLOW_USE_SYCL + template <> struct SetZeroFunctor<Eigen::ThreadPoolDevice, string> { void operator()(const Eigen::ThreadPoolDevice& d, |