diff options
author | RJ Ryan <rjryan@google.com> | 2018-06-27 18:11:03 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-27 18:13:25 -0700 |
commit | 128b69878362dcec736326d52828d2167ce291cb (patch) | |
tree | 7cc93883a8248c3ee8f799502e4d39646a31a419 /tensorflow/contrib/lite/schema | |
parent | 2b3b5054c7ceff0bc2811cfe0ebc063947801ce0 (diff) |
Add complex64 support to tf.lite runtime.
PiperOrigin-RevId: 202403235
Diffstat (limited to 'tensorflow/contrib/lite/schema')
-rw-r--r-- | tensorflow/contrib/lite/schema/schema.fbs | 1 | ||||
-rwxr-xr-x | tensorflow/contrib/lite/schema/schema_generated.h | 9 |
2 files changed, 7 insertions, 3 deletions
diff --git a/tensorflow/contrib/lite/schema/schema.fbs b/tensorflow/contrib/lite/schema/schema.fbs index 76ad3ef893..15fb8bbdb8 100644 --- a/tensorflow/contrib/lite/schema/schema.fbs +++ b/tensorflow/contrib/lite/schema/schema.fbs @@ -35,6 +35,7 @@ enum TensorType : byte { STRING = 5, BOOL = 6, INT16 = 7, + COMPLEX64 = 8, } // Parameters for converting a quantized tensor back to float. Given a diff --git a/tensorflow/contrib/lite/schema/schema_generated.h b/tensorflow/contrib/lite/schema/schema_generated.h index e3ce90aa55..fe0ff9a7a5 100755 --- a/tensorflow/contrib/lite/schema/schema_generated.h +++ b/tensorflow/contrib/lite/schema/schema_generated.h @@ -223,11 +223,12 @@ enum TensorType { TensorType_STRING = 5, TensorType_BOOL = 6, TensorType_INT16 = 7, + TensorType_COMPLEX64 = 8, TensorType_MIN = TensorType_FLOAT32, - TensorType_MAX = TensorType_INT16 + TensorType_MAX = TensorType_COMPLEX64 }; -inline TensorType (&EnumValuesTensorType())[8] { +inline TensorType (&EnumValuesTensorType())[9] { static TensorType values[] = { TensorType_FLOAT32, TensorType_FLOAT16, @@ -236,7 +237,8 @@ inline TensorType (&EnumValuesTensorType())[8] { TensorType_INT64, TensorType_STRING, TensorType_BOOL, - TensorType_INT16 + TensorType_INT16, + TensorType_COMPLEX64 }; return values; } @@ -251,6 +253,7 @@ inline const char **EnumNamesTensorType() { "STRING", "BOOL", "INT16", + "COMPLEX64", nullptr }; return names; |