diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-13 13:24:52 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-13 13:27:55 -0700 |
| commit | b253e6b874d4f4d242b5d31777462cac146935d2 (patch) | |
| tree | 8cf29928c90c20ea14c1cf665b1b8f80f6ec3a34 /tensorflow/contrib/lite/context.h | |
| parent | 91034421a2422c24a177b8d4a46f9fc3d157be3f (diff) | |
support int16-quantized data in TFLite interpreter.
PiperOrigin-RevId: 200442886
Diffstat (limited to 'tensorflow/contrib/lite/context.h')
| -rw-r--r-- | tensorflow/contrib/lite/context.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/context.h b/tensorflow/contrib/lite/context.h index 4eb66cc225..0415acfe0f 100644 --- a/tensorflow/contrib/lite/context.h +++ b/tensorflow/contrib/lite/context.h @@ -138,6 +138,7 @@ typedef enum { kTfLiteInt64 = 4, kTfLiteString = 5, kTfLiteBool = 6, + kTfLiteInt16 = 7, } TfLiteType; // Parameters for asymmetric quantization. Quantized values can be converted @@ -148,7 +149,7 @@ typedef struct { int32_t zero_point; } TfLiteQuantizationParams; -// A union of points that points to memory for a given tensor. +// A union of pointers that points to memory for a given tensor. typedef union { int* i32; int64_t* i64; @@ -157,6 +158,7 @@ typedef union { const char* raw_const; uint8_t* uint8; bool* b; + int16_t* i16; } TfLitePtrUnion; // Memory allocation strategies. kTfLiteMmapRo is for read-only memory-mapped |