Fix a case where DT_HALF would not be part of the right type lists

for __ANDROID_TYPES_FULL__.
Change: 117936401

1 files changed, 4 insertions, 3 deletions

diff --git a/tensorflow/core/framework/types.cc b/tensorflow/core/framework/types.cc
index fceb2f3a04..5516a09a98 100644
--- a/tensorflow/core/framework/types.cc
+++ b/tensorflow/core/framework/types.cc
@@ -209,7 +209,8 @@ DataTypeVector NumberTypes() {
 DataTypeVector RealNumberTypes() { return {DT_FLOAT, DT_INT32, DT_INT64}; }
 
 DataTypeVector NumberTypes() {
-  return {DT_FLOAT, DT_INT32, DT_INT64, DT_QINT8, DT_QUINT8, DT_QINT32};
+  return {DT_FLOAT,  DT_INT32,  DT_INT64, DT_QINT8,
+          DT_QUINT8, DT_QINT32, DT_HALF};
 }
 
 DataTypeVector QuantizedTypes() {
@@ -217,8 +218,8 @@ DataTypeVector QuantizedTypes() {
 }
 
 DataTypeVector RealAndQuantizedTypes() {
-  return {DT_FLOAT,  DT_INT32,  DT_INT64,   DT_QINT8,
-          DT_QUINT8, DT_QINT16, DT_QUINT16, DT_QINT32};
+  return {DT_FLOAT,  DT_INT32,   DT_INT64,  DT_QINT8, DT_QUINT8,
+          DT_QINT16, DT_QUINT16, DT_QINT32, DT_HALF};
 }
 
 #else  // defined(__ANDROID__) && !defined(__ANDROID_TYPES_FULL__)