diff options
Diffstat (limited to 'tensorflow/core/kernels/tensor_array_ops.cc')
| -rw-r--r-- | tensorflow/core/kernels/tensor_array_ops.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/tensor_array_ops.cc b/tensorflow/core/kernels/tensor_array_ops.cc index 37803ec775..5aa5d20b1a 100644 --- a/tensorflow/core/kernels/tensor_array_ops.cc +++ b/tensorflow/core/kernels/tensor_array_ops.cc @@ -735,6 +735,7 @@ class TensorArrayPackOrGatherOp : public OpKernel { TensorArrayPackOrGatherOp<CPUDevice, type, false /* LEGACY_PACK */>); TF_CALL_POD_STRING_TYPES(REGISTER_GATHER_AND_PACK); +TF_CALL_variant(REGISTER_GATHER_AND_PACK); REGISTER_GATHER_AND_PACK(quint8); REGISTER_GATHER_AND_PACK(qint8); REGISTER_GATHER_AND_PACK(qint32); |