aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/schema
diff options
context:
space:
mode:
authorGravatar RJ Ryan <rjryan@google.com>2018-06-27 18:11:03 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-06-27 18:13:25 -0700
commit128b69878362dcec736326d52828d2167ce291cb (patch)
tree7cc93883a8248c3ee8f799502e4d39646a31a419 /tensorflow/contrib/lite/schema
parent2b3b5054c7ceff0bc2811cfe0ebc063947801ce0 (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.fbs1
-rwxr-xr-xtensorflow/contrib/lite/schema/schema_generated.h9
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;