diff options
Diffstat (limited to 'tensorflow/core/kernels/tile_ops_cpu_impl.h')
| -rw-r--r-- | tensorflow/core/kernels/tile_ops_cpu_impl.h | 35 |
1 files changed, 1 insertions, 34 deletions
diff --git a/tensorflow/core/kernels/tile_ops_cpu_impl.h b/tensorflow/core/kernels/tile_ops_cpu_impl.h index db3f046439..a6eed4935d 100644 --- a/tensorflow/core/kernels/tile_ops_cpu_impl.h +++ b/tensorflow/core/kernels/tile_ops_cpu_impl.h @@ -21,29 +21,11 @@ limitations under the License. #include "tensorflow/core/kernels/tile_ops_impl.h" namespace tensorflow { + namespace functor { typedef Eigen::ThreadPoolDevice CPUDevice; -// Register functors used for TileOp. -#define DEFINE_DIM(T, NDIM) template struct Tile<CPUDevice, T, NDIM>; -#define DEFINE_TYPE(T) DEFINE_DIM(T, CPU_PROVIDED_IXDIM) - -TF_CALL_bool(DEFINE_TYPE); -TF_CALL_float(DEFINE_TYPE); -TF_CALL_double(DEFINE_TYPE); -TF_CALL_uint8(DEFINE_TYPE); -TF_CALL_int32(DEFINE_TYPE); -TF_CALL_int16(DEFINE_TYPE); -TF_CALL_int64(DEFINE_TYPE); -TF_CALL_half(DEFINE_TYPE); -TF_CALL_complex64(DEFINE_TYPE); -TF_CALL_complex128(DEFINE_TYPE); -TF_CALL_string(DEFINE_TYPE); - -#undef DEFINE_DIM -#undef DEFINE_TYPE - // Register functors used for TileGradientOp. #define DEFINE_DIM(T, NDIM) \ template struct TileGrad<CPUDevice, T, NDIM>; \ @@ -65,21 +47,6 @@ TF_CALL_complex128(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_bool(DEFINE_TYPE); -TF_CALL_float(DEFINE_TYPE); -TF_CALL_double(DEFINE_TYPE); -TF_CALL_uint8(DEFINE_TYPE); -TF_CALL_int32(DEFINE_TYPE); -TF_CALL_int16(DEFINE_TYPE); -TF_CALL_int64(DEFINE_TYPE); - -#undef DEFINE_DIM -#undef DEFINE_TYPE - // Register functors used for TileGradientOp. #define DEFINE_DIM(T, NDIM) \ template struct TileGrad<SYCLDevice, T, NDIM>; \ |