diff options
author | 2017-11-24 17:50:04 -0800 | |
---|---|---|
committer | 2017-11-24 18:05:02 -0800 | |
commit | 3e6edce1f41a79ca83358b14af9230826e871b66 (patch) | |
tree | 85129cd4f934ab4fc2599e7d77a53a3bd8ddbedc | |
parent | 1d77785e9e13241cb318edce4661e0bdc2dd3095 (diff) |
Address `Eigen::Half` -> `Eigen::half`
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
-rw-r--r-- | tensorflow/core/kernels/spacetodepth_op.cc | 4 | ||||
-rw-r--r-- | tensorflow/core/kernels/spacetodepth_op_gpu.cu.cc | 6 |
2 files changed, 5 insertions, 5 deletions
diff --git a/tensorflow/core/kernels/spacetodepth_op.cc b/tensorflow/core/kernels/spacetodepth_op.cc index d93a2a9bad..e59adfc6ac 100644 --- a/tensorflow/core/kernels/spacetodepth_op.cc +++ b/tensorflow/core/kernels/spacetodepth_op.cc @@ -188,8 +188,8 @@ REGISTER_KERNEL_BUILDER( Name("SpaceToDepth").Device(DEVICE_GPU).TypeConstraint<float>("T"), SpaceToDepthOp<GPUDevice, float>); REGISTER_KERNEL_BUILDER( - Name("SpaceToDepth").Device(DEVICE_GPU).TypeConstraint<Eigen::Half>("T"), - SpaceToDepthOp<GPUDevice, Eigen::Half>); + Name("SpaceToDepth").Device(DEVICE_GPU).TypeConstraint<Eigen::half>("T"), + SpaceToDepthOp<GPUDevice, Eigen::half>); REGISTER_KERNEL_BUILDER( Name("SpaceToDepth").Device(DEVICE_GPU).TypeConstraint<qint8>("T"), SpaceToDepthOp<GPUDevice, qint8>); diff --git a/tensorflow/core/kernels/spacetodepth_op_gpu.cu.cc b/tensorflow/core/kernels/spacetodepth_op_gpu.cu.cc index e841472972..8466fa192f 100644 --- a/tensorflow/core/kernels/spacetodepth_op_gpu.cu.cc +++ b/tensorflow/core/kernels/spacetodepth_op_gpu.cu.cc @@ -225,9 +225,9 @@ struct SpaceToDepthOpFunctor<GPUDevice, T, FORMAT_NCHW> { template struct functor::SpaceToDepthOpFunctor<GPUDevice, float, FORMAT_NCHW>; template struct functor::SpaceToDepthOpFunctor<GPUDevice, float, FORMAT_NHWC>; -// Instantiate the GPU implementations for Eigen::Half. -template struct functor::SpaceToDepthOpFunctor<GPUDevice, Eigen::Half, FORMAT_NCHW>; -template struct functor::SpaceToDepthOpFunctor<GPUDevice, EIgen::Half, FORMAT_NHWC>; +// Instantiate the GPU implementations for Eigen::half. +template struct functor::SpaceToDepthOpFunctor<GPUDevice, Eigen::half, FORMAT_NCHW>; +template struct functor::SpaceToDepthOpFunctor<GPUDevice, Eigen::half, FORMAT_NHWC>; // NCHW_VECT_C with 4 x qint8 can be treated as NCHW int32. template struct functor::SpaceToDepthOpFunctor<GPUDevice, int32, FORMAT_NCHW>; |