diff options
Diffstat (limited to 'tensorflow/core/framework/types.cc')
| -rw-r--r-- | tensorflow/core/framework/types.cc | 18 |
1 files changed, 13 insertions, 5 deletions
diff --git a/tensorflow/core/framework/types.cc b/tensorflow/core/framework/types.cc index 54b55e49c0..c87a0445cd 100644 --- a/tensorflow/core/framework/types.cc +++ b/tensorflow/core/framework/types.cc @@ -64,6 +64,8 @@ string DataTypeString(DataType dtype) { return "string"; case DT_COMPLEX64: return "complex64"; + case DT_COMPLEX128: + return "complex128"; case DT_INT64: return "int64"; case DT_BOOL: @@ -125,6 +127,9 @@ bool DataTypeFromString(StringPiece sp, DataType* dt) { } else if (sp == "complex64") { *dt = DT_COMPLEX64; return true; + } else if (sp == "complex128") { + *dt = DT_COMPLEX128; + return true; } else if (sp == "int64") { *dt = DT_INT64; return true; @@ -165,9 +170,10 @@ string DataTypeSliceString(const DataTypeSlice types) { } DataTypeVector AllTypes() { - return {DT_FLOAT, DT_DOUBLE, DT_INT32, DT_UINT8, DT_INT16, DT_UINT16, - DT_INT8, DT_STRING, DT_COMPLEX64, DT_INT64, DT_BOOL, DT_QINT8, - DT_QUINT8, DT_QINT16, DT_QUINT16, DT_QINT32}; + return {DT_FLOAT, DT_DOUBLE, DT_INT32, DT_UINT8, DT_INT16, + DT_UINT16, DT_INT8, DT_STRING, DT_COMPLEX64, DT_COMPLEX128, + DT_INT64, DT_BOOL, DT_QINT8, DT_QUINT8, DT_QINT16, + DT_QUINT16, DT_QINT32}; } #if !defined(__ANDROID__) @@ -188,8 +194,9 @@ DataTypeVector RealAndQuantizedTypes() { } DataTypeVector NumberTypes() { - return {DT_FLOAT, DT_DOUBLE, DT_INT64, DT_INT32, DT_UINT8, DT_UINT16, - DT_INT16, DT_INT8, DT_COMPLEX64, DT_QINT8, DT_QUINT8, DT_QINT32}; + return {DT_FLOAT, DT_DOUBLE, DT_INT64, DT_INT32, DT_UINT8, + DT_UINT16, DT_INT16, DT_INT8, DT_COMPLEX64, DT_COMPLEX128, + DT_QINT8, DT_QUINT8, DT_QINT32 }; } #else // defined(__ANDROID__) @@ -223,6 +230,7 @@ bool DataTypeCanUseMemcpy(DataType dt) { case DT_INT16: case DT_INT8: case DT_COMPLEX64: + case DT_COMPLEX128: case DT_INT64: case DT_BOOL: case DT_QINT8: |