diff options
Diffstat (limited to 'tensorflow/core/kernels/conv_ops_gpu_3.cu.cc')
-rw-r--r-- | tensorflow/core/kernels/conv_ops_gpu_3.cu.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/conv_ops_gpu_3.cu.cc b/tensorflow/core/kernels/conv_ops_gpu_3.cu.cc index bb99d627a5..2307c2de0e 100644 --- a/tensorflow/core/kernels/conv_ops_gpu_3.cu.cc +++ b/tensorflow/core/kernels/conv_ops_gpu_3.cu.cc @@ -548,9 +548,11 @@ template struct functor::TransformFilter<GPUDevice, Eigen::half, int, 4>; template struct functor::ReverseTransformFilter<GPUDevice, float, 4>; template struct functor::ReverseTransformFilter<GPUDevice, Eigen::half, 4>; +template struct functor::NHWCToNCHW<GPUDevice, double, 4>; template struct functor::NHWCToNCHW<GPUDevice, float, 4>; template struct functor::NHWCToNCHW<GPUDevice, Eigen::half, 4>; +template struct functor::NCHWToNHWC<GPUDevice, double, 4>; template struct functor::NCHWToNHWC<GPUDevice, float, 4>; template struct functor::NCHWToNHWC<GPUDevice, Eigen::half, 4>; |