diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-09 11:14:25 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-09 11:16:51 -0700 |
| commit | 20e1f3a852c83d9369c7d56a943fe6b8f9b88644 (patch) | |
| tree | f54acb82a9a7a5b6324ee0a3e21cb6025b84b961 /tensorflow/core/kernels/concat_lib_gpu.cc | |
| parent | 1b97ea722cea69a8d9a6fe3bd515e22f356d40ab (diff) | |
Register tf.concat with uint8 data type.
PiperOrigin-RevId: 192154998
Diffstat (limited to 'tensorflow/core/kernels/concat_lib_gpu.cc')
| -rw-r--r-- | tensorflow/core/kernels/concat_lib_gpu.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/concat_lib_gpu.cc b/tensorflow/core/kernels/concat_lib_gpu.cc index d8643c0b2f..93e392d303 100644 --- a/tensorflow/core/kernels/concat_lib_gpu.cc +++ b/tensorflow/core/kernels/concat_lib_gpu.cc @@ -118,6 +118,7 @@ TF_CALL_complex128(REGISTER); TF_CALL_int64(REGISTER); TF_CALL_bfloat16(REGISTER); TF_CALL_bool(REGISTER); +TF_CALL_uint8(REGISTER); #undef REGISTER |