summaryrefslogtreecommitdiff
path: root/compiler/tflite2circle/src/TFLTensorType.lst
diff options
context:
space:
mode:
Diffstat (limited to 'compiler/tflite2circle/src/TFLTensorType.lst')
-rw-r--r--compiler/tflite2circle/src/TFLTensorType.lst16
1 files changed, 16 insertions, 0 deletions
diff --git a/compiler/tflite2circle/src/TFLTensorType.lst b/compiler/tflite2circle/src/TFLTensorType.lst
new file mode 100644
index 000000000..968153a5d
--- /dev/null
+++ b/compiler/tflite2circle/src/TFLTensorType.lst
@@ -0,0 +1,16 @@
+#ifndef TFL_TENSORTYPE
+#error "TFL_TENSORTYPE is not defined."
+#endif
+
+// NOTE order of type follows TensorFlow lite schema.fbs 'enum TensorType'
+
+TFL_TENSORTYPE(FLOAT32)
+TFL_TENSORTYPE(FLOAT16)
+TFL_TENSORTYPE(INT32)
+TFL_TENSORTYPE(UINT8)
+TFL_TENSORTYPE(INT64)
+TFL_TENSORTYPE(STRING)
+TFL_TENSORTYPE(BOOL)
+TFL_TENSORTYPE(INT16)
+TFL_TENSORTYPE(COMPLEX64)
+TFL_TENSORTYPE(INT8)