diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-17 07:41:09 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-17 07:43:45 -0700 |
| commit | fb7675e06d6b5ee1d45dcd4eda64a4caa689e393 (patch) | |
| tree | 7bc99314abe91bc7703a3739d004f61373bde470 /tensorflow/core/kernels/gather_op.cc | |
| parent | f73d793e7a9234efb14fd8f11322429d122949b1 (diff) | |
Add uint32/uint64 support to Gather op.
PiperOrigin-RevId: 193195939
Diffstat (limited to 'tensorflow/core/kernels/gather_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/gather_op.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/gather_op.cc b/tensorflow/core/kernels/gather_op.cc index 08adf4badb..ef332ebee3 100644 --- a/tensorflow/core/kernels/gather_op.cc +++ b/tensorflow/core/kernels/gather_op.cc @@ -143,6 +143,8 @@ TF_CALL_ALL_TYPES(REGISTER_GATHER_CPU); TF_CALL_QUANTIZED_TYPES(REGISTER_GATHER_CPU); TF_CALL_quint16(REGISTER_GATHER_CPU); TF_CALL_qint16(REGISTER_GATHER_CPU); +TF_CALL_uint32(REGISTER_GATHER_CPU); +TF_CALL_uint64(REGISTER_GATHER_CPU); #undef REGISTER_GATHER_CPU |