diff options
author | 2016-07-25 13:48:16 -0800 | |
---|---|---|
committer | 2016-07-25 15:02:34 -0700 | |
commit | 21716d8f6e175cd6e8cd97a84e48497574268b0c (patch) | |
tree | 3345202e7a812cc9572beb24fc01732696a4140d /tensorflow/core/kernels/reverse_op.cc | |
parent | ed281973d66d0030e58a77a05821bbb88627f5bd (diff) |
Merge changes from github.
Change: 128401884
Diffstat (limited to 'tensorflow/core/kernels/reverse_op.cc')
-rw-r--r-- | tensorflow/core/kernels/reverse_op.cc | 16 |
1 files changed, 9 insertions, 7 deletions
diff --git a/tensorflow/core/kernels/reverse_op.cc b/tensorflow/core/kernels/reverse_op.cc index 18fb480515..30e90f2206 100644 --- a/tensorflow/core/kernels/reverse_op.cc +++ b/tensorflow/core/kernels/reverse_op.cc @@ -97,13 +97,7 @@ class ReverseOp : public OpKernel { .HostMemory("dims"), \ ReverseOp<CPUDevice, T>) -TF_CALL_uint8(REGISTER_KERNEL); -TF_CALL_int8(REGISTER_KERNEL); -TF_CALL_int32(REGISTER_KERNEL); -TF_CALL_bool(REGISTER_KERNEL); -TF_CALL_half(REGISTER_KERNEL); -TF_CALL_float(REGISTER_KERNEL); -TF_CALL_double(REGISTER_KERNEL); +TF_CALL_POD_TYPES(REGISTER_KERNEL); #undef REGISTER_KERNEL #if GOOGLE_CUDA @@ -136,6 +130,8 @@ TF_CALL_bool(DECLARE_GPU_SPEC); TF_CALL_half(DECLARE_GPU_SPEC); TF_CALL_float(DECLARE_GPU_SPEC); TF_CALL_double(DECLARE_GPU_SPEC); +TF_CALL_complex64(DECLARE_GPU_SPEC); +TF_CALL_complex128(DECLARE_GPU_SPEC); #undef DECLARE_GPU_SPEC #undef DECLARE_GPU_SPEC_DIM } // namespace functor @@ -149,9 +145,15 @@ TF_CALL_double(DECLARE_GPU_SPEC); ReverseOp<GPUDevice, T>) TF_CALL_uint8(REGISTER_GPU_KERNEL); TF_CALL_int8(REGISTER_GPU_KERNEL); +// TODO Find out why the int32 GPU kernel doesn't work +// and decide whether we want to enable the bool kernel. +//TF_CALL_int32(REGISTER_GPU_KERNEL); +//TF_CALL_bool(REGISTER_GPU_KERNEL); TF_CALL_half(REGISTER_GPU_KERNEL); TF_CALL_float(REGISTER_GPU_KERNEL); TF_CALL_double(REGISTER_GPU_KERNEL); +TF_CALL_complex64(REGISTER_GPU_KERNEL); +TF_CALL_complex128(REGISTER_GPU_KERNEL); #undef REGISTER_GPU_KERNEL #endif // GOOGLE_CUDA |