aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/util/saved_tensor_slice_util.h
diff options
context:
space:
mode:
authorGravatar Ben Lee <blee@google.com>2018-06-25 13:51:48 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-06-25 13:54:05 -0700
commit3839438f3910450c3361122a5745ff580e8c3292 (patch)
treec27064b4a20d8f95d248f41f1a12b259d42b16bc /tensorflow/core/util/saved_tensor_slice_util.h
parent4c8501a86d6df11c901e4be8e7596a29165a2225 (diff)
Support quint16 in Assign on CPU.
- Add missing TENSOR_PROTO_EXTRACT_TYPE for quint16 PiperOrigin-RevId: 202013023
Diffstat (limited to 'tensorflow/core/util/saved_tensor_slice_util.h')
-rw-r--r--tensorflow/core/util/saved_tensor_slice_util.h1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/util/saved_tensor_slice_util.h b/tensorflow/core/util/saved_tensor_slice_util.h
index ee43945a39..90672a10a8 100644
--- a/tensorflow/core/util/saved_tensor_slice_util.h
+++ b/tensorflow/core/util/saved_tensor_slice_util.h
@@ -123,6 +123,7 @@ TENSOR_PROTO_EXTRACT_TYPE(int8, int, int32);
TENSOR_PROTO_EXTRACT_TYPE(int16, int, int32);
TENSOR_PROTO_EXTRACT_TYPE(qint8, int, int32);
TENSOR_PROTO_EXTRACT_TYPE(quint8, int, int32);
+TENSOR_PROTO_EXTRACT_TYPE(quint16, int, int32);
#undef TENSOR_PROTO_EXTRACT_TYPE_COMPLEX
#undef TENSOR_PROTO_EXTRACT_TYPE_HELPER