diff options
| author |  CQY <qychen@pku.edu.cn> | 2017-10-15 11:04:45 +0800 |
|---|---|---|
| committer | CQY <qychen@pku.edu.cn> | 2017-10-15 11:04:45 +0800 |
| commit | e49839df71904f6f7f22245f9eaaeb6487c9d196 (patch) | |
| tree | 4301af48b5e9af2ea90be9e95be9f6672d06b81a /tensorflow/core/kernels/reverse_sequence_op.cc | |
| parent | 7cdd26f606b39e3e487ec15dfa6eb5c6cf63ef84 (diff) | |
Support reversing bool sequence.
Diffstat (limited to 'tensorflow/core/kernels/reverse_sequence_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/reverse_sequence_op.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/reverse_sequence_op.cc b/tensorflow/core/kernels/reverse_sequence_op.cc index 505c512cc4..d1980d4b65 100644 --- a/tensorflow/core/kernels/reverse_sequence_op.cc +++ b/tensorflow/core/kernels/reverse_sequence_op.cc @@ -175,6 +175,7 @@ class ReverseSequenceOp : public OpKernel { REGISTER_REVERSE_SEQUENCE(type, int64); TF_CALL_NUMBER_TYPES(REGISTER_REVERSE_SEQUENCE_LEN); +TF_CALL_bool(REGISTER_REVERSE_SEQUENCE_LEN); #if GOOGLE_CUDA @@ -200,6 +201,7 @@ namespace functor { DECLARE_GPU_SPEC_LEN(T, 5); TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS); +TF_CALL_bool(DECLARE_GPU_SPECS); } // namespace functor @@ -215,6 +217,7 @@ TF_CALL_GPU_NUMBER_TYPES(DECLARE_GPU_SPECS); REGISTER_REVERSE_SEQUENCE_GPU(type, int64); TF_CALL_GPU_NUMBER_TYPES(REGISTER_REVERSE_SEQUENCE_GPU_LEN); +TF_CALL_bool(REGISTER_REVERSE_SEQUENCE_GPU_LEN); #undef REGISTER_REVERSE_SEQUENCE_GPU |