diff options
Diffstat (limited to 'tensorflow/core/kernels/tile_ops_cpu_impl.h')
| -rw-r--r-- | tensorflow/core/kernels/tile_ops_cpu_impl.h | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/tile_ops_cpu_impl.h b/tensorflow/core/kernels/tile_ops_cpu_impl.h index 9cdf69ad0b..650c739ed5 100644 --- a/tensorflow/core/kernels/tile_ops_cpu_impl.h +++ b/tensorflow/core/kernels/tile_ops_cpu_impl.h @@ -62,6 +62,30 @@ TF_CALL_complex128(DEFINE_TYPE); #undef DEFINE_DIM #undef DEFINE_TYPE +#ifdef TENSORFLOW_USE_SYCL +typedef Eigen::SyclDevice SYCLDevice; + +// Register functors used for TileOp. +#define DEFINE_DIM(T, NDIM) template struct Tile<SYCLDevice, T, NDIM>; +#define DEFINE_TYPE(T) DEFINE_DIM(T, CPU_PROVIDED_IXDIM) + +TF_CALL_float(DEFINE_TYPE); + +#undef DEFINE_DIM +#undef DEFINE_TYPE + +// Register functors used for TileGradientOp. +#define DEFINE_DIM(T, NDIM) \ + template struct TileGrad<SYCLDevice, T, NDIM>; \ + template struct ReduceAndReshape<SYCLDevice, T, NDIM, 1>; +#define DEFINE_TYPE(T) DEFINE_DIM(T, CPU_PROVIDED_IXDIM) + +TF_CALL_float(DEFINE_TYPE); + +#undef DEFINE_DIM +#undef DEFINE_TYPE +#endif // TENSORFLOW_USE_SYCL + } // end namespace functor } // end namespace tensorflow |