diff options
| author |  Lakshay Garg <lakshay.garg.1996@gmail.com> | 2017-06-09 12:02:13 +0530 |
|---|---|---|
| committer |  Martin Wicke <martin.wicke@gmail.com> | 2017-06-14 00:08:30 -0700 |
| commit | 4c0052dc4b7c49a876166113b49960a57f7db939 (patch) | |
| tree | 2becc6f2ea0589c941ece18e0a624704f404a721 /tensorflow/core/kernels/cwise_op_gpu_add.cu.cc | |
| parent | 3c4cb087e6212a0dbc930237fbc49a7c715da97d (diff) | |
Added uint8 registration for addition operation (Fixes #10447)
Diffstat (limited to 'tensorflow/core/kernels/cwise_op_gpu_add.cu.cc')
| -rw-r--r-- | tensorflow/core/kernels/cwise_op_gpu_add.cu.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/cwise_op_gpu_add.cu.cc b/tensorflow/core/kernels/cwise_op_gpu_add.cu.cc index 5aaf2b5b4b..61079ebab3 100644 --- a/tensorflow/core/kernels/cwise_op_gpu_add.cu.cc +++ b/tensorflow/core/kernels/cwise_op_gpu_add.cu.cc @@ -19,7 +19,8 @@ limitations under the License. namespace tensorflow { namespace functor { -DEFINE_BINARY6(add, Eigen::half, float, double, int64, complex64, complex128); +DEFINE_BINARY7(add, Eigen::half, float, double, uint8, int64, complex64, + complex128); } // namespace functor } // namespace tensorflow |