aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/context.h
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-04-12 19:52:18 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-04-12 19:54:58 -0700
commit3c9870524b86fe7e3cff5a49daa692cd52e7f0c4 (patch)
tree670d4512bf9ef3df932eda5eb383033bc2d0dd4a /tensorflow/contrib/lite/context.h
parent4f615adc1d7875f9fbe592619dc6b0f31cc7fd9e (diff)
Add boolean type to tflite in favor of comparison implementations.
PiperOrigin-RevId: 192711203
Diffstat (limited to 'tensorflow/contrib/lite/context.h')
-rw-r--r--tensorflow/contrib/lite/context.h2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/context.h b/tensorflow/contrib/lite/context.h
index 45184b05ec..0b38f43cd3 100644
--- a/tensorflow/contrib/lite/context.h
+++ b/tensorflow/contrib/lite/context.h
@@ -137,6 +137,7 @@ typedef enum {
kTfLiteUInt8 = 3,
kTfLiteInt64 = 4,
kTfLiteString = 5,
+ kTfLiteBool = 6,
} TfLiteType;
// Parameters for asymmetric quantization. Quantized values can be converted
@@ -155,6 +156,7 @@ typedef union {
char* raw;
const char* raw_const;
uint8_t* uint8;
+ bool* b;
} TfLitePtrUnion;
// Memory allocation strategies. kTfLiteMmapRo is for read-only memory-mapped