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_op.cc | |
| parent | 1b97ea722cea69a8d9a6fe3bd515e22f356d40ab (diff) | |
Register tf.concat with uint8 data type.
PiperOrigin-RevId: 192154998
Diffstat (limited to 'tensorflow/core/kernels/concat_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/concat_op.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/concat_op.cc b/tensorflow/core/kernels/concat_op.cc index f16766315f..a87b63f913 100644 --- a/tensorflow/core/kernels/concat_op.cc +++ b/tensorflow/core/kernels/concat_op.cc @@ -212,6 +212,7 @@ REGISTER_CONCAT(qint32); TF_CALL_GPU_NUMBER_TYPES(REGISTER_GPU); REGISTER_GPU(bfloat16); +TF_CALL_uint8(REGISTER_GPU); TF_CALL_complex64(REGISTER_GPU); TF_CALL_complex128(REGISTER_GPU); TF_CALL_int64(REGISTER_GPU); |