diff options
Diffstat (limited to 'tensorflow/core/kernels/batchtospace_op.cc')
-rw-r--r-- | tensorflow/core/kernels/batchtospace_op.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/batchtospace_op.cc b/tensorflow/core/kernels/batchtospace_op.cc index 8a2c5e21ac..277f90cdad 100644 --- a/tensorflow/core/kernels/batchtospace_op.cc +++ b/tensorflow/core/kernels/batchtospace_op.cc @@ -275,7 +275,7 @@ TF_CALL_REAL_NUMBER_TYPES(REGISTER); .HostMemory("crops"), \ BatchToSpaceOp<GPUDevice, T>); -TF_CALL_GPU_NUMBER_TYPES_NO_HALF(REGISTER); +TF_CALL_GPU_NUMBER_TYPES(REGISTER); #undef REGISTER #endif // GOOGLE_CUDA |