diff options
| author |  Ziming Dong <dzm1016397507@gmail.com> | 2016-08-23 13:46:11 -0400 |
|---|---|---|
| committer |  Rasmus Munk Larsen <rmlarsen@google.com> | 2016-08-23 10:46:11 -0700 |
| commit | 09999037cdbbea183ec4e85825f5b06b54745783 (patch) | |
| tree | 637e20bfcc7fa6da00ee6d2c2ded2f9630f1e313 /tensorflow/core/kernels/reverse_sequence_op_gpu.cu.cc | |
| parent | 459c2fed498530b794c4871892fd68d1e6834ac6 (diff) | |
add int32 support for param seq_length of op reverse_sequence (#3774)
* add int32 support for seq_length in reverse_sequence
* modify gpu version of reverse_sequence
* fix bugs
* fix backwards_compatibility
Diffstat (limited to 'tensorflow/core/kernels/reverse_sequence_op_gpu.cu.cc')
| -rw-r--r-- | tensorflow/core/kernels/reverse_sequence_op_gpu.cu.cc | 18 |
1 files changed, 11 insertions, 7 deletions
diff --git a/tensorflow/core/kernels/reverse_sequence_op_gpu.cu.cc b/tensorflow/core/kernels/reverse_sequence_op_gpu.cu.cc index bcc265c624..373fd60687 100644 --- a/tensorflow/core/kernels/reverse_sequence_op_gpu.cu.cc +++ b/tensorflow/core/kernels/reverse_sequence_op_gpu.cu.cc @@ -24,15 +24,19 @@ namespace tensorflow { typedef Eigen::GpuDevice GPUDevice; -#define DEFINE_GPU_SPEC(T, dims) \ - template class generator::ReverseGenerator<T, dims>; \ - template struct functor::ReverseSequence<GPUDevice, T, dims>; +#define DEFINE_GPU_SPEC(T, Tlen, dims) \ + template class generator::ReverseGenerator<T, Tlen, dims>; \ + template struct functor::ReverseSequence<GPUDevice, T, Tlen, dims>; + +#define DEFINE_GPU_SPEC_LEN(T, dims) \ + DEFINE_GPU_SPEC(T, int32, dims); \ + DEFINE_GPU_SPEC(T, int64, dims); #define DEFINE_GPU_SPECS(T) \ - DEFINE_GPU_SPEC(T, 2); \ - DEFINE_GPU_SPEC(T, 3); \ - DEFINE_GPU_SPEC(T, 4); \ - DEFINE_GPU_SPEC(T, 5); + DEFINE_GPU_SPEC_LEN(T, 2); \ + DEFINE_GPU_SPEC_LEN(T, 3); \ + DEFINE_GPU_SPEC_LEN(T, 4); \ + DEFINE_GPU_SPEC_LEN(T, 5); TF_CALL_GPU_NUMBER_TYPES(DEFINE_GPU_SPECS); |