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/model.cc | |
| parent | 91034421a2422c24a177b8d4a46f9fc3d157be3f (diff) | |
support int16-quantized data in TFLite interpreter.
PiperOrigin-RevId: 200442886
Diffstat (limited to 'tensorflow/contrib/lite/model.cc')
| -rw-r--r-- | tensorflow/contrib/lite/model.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/model.cc b/tensorflow/contrib/lite/model.cc index 039f32b38e..cd7b9bdabf 100644 --- a/tensorflow/contrib/lite/model.cc +++ b/tensorflow/contrib/lite/model.cc @@ -45,6 +45,9 @@ TfLiteStatus ConvertTensorType(TensorType tensor_type, TfLiteType* type, case TensorType_FLOAT32: *type = kTfLiteFloat32; break; + case TensorType_INT16: + *type = kTfLiteInt16; + break; case TensorType_INT32: *type = kTfLiteInt32; break; |