diff options
Diffstat (limited to 'tensorflow/core/kernels/depthtospace_op_gpu.cu.cc')
| -rw-r--r-- | tensorflow/core/kernels/depthtospace_op_gpu.cu.cc | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/depthtospace_op_gpu.cu.cc b/tensorflow/core/kernels/depthtospace_op_gpu.cu.cc index 184c703599..0656081177 100644 --- a/tensorflow/core/kernels/depthtospace_op_gpu.cu.cc +++ b/tensorflow/core/kernels/depthtospace_op_gpu.cu.cc @@ -238,6 +238,12 @@ struct DepthToSpaceOpFunctor<GPUDevice, T, FORMAT_NCHW> { template struct functor::DepthToSpaceOpFunctor<GPUDevice, float, FORMAT_NCHW>; template struct functor::DepthToSpaceOpFunctor<GPUDevice, float, FORMAT_NHWC>; +// Instantiate the GPU implementations for Eigen::half. +template struct functor::DepthToSpaceOpFunctor<GPUDevice, Eigen::half, + FORMAT_NCHW>; +template struct functor::DepthToSpaceOpFunctor<GPUDevice, Eigen::half, + FORMAT_NHWC>; + // NCHW_VECT_C with 4 x qint8 can be treated as NCHW int32. template struct functor::DepthToSpaceOpFunctor<GPUDevice, int32, FORMAT_NCHW>; |