diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-12 19:52:18 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-12 19:54:58 -0700 |
commit | 3c9870524b86fe7e3cff5a49daa692cd52e7f0c4 (patch) | |
tree | 670d4512bf9ef3df932eda5eb383033bc2d0dd4a /tensorflow/contrib/lite/interpreter.cc | |
parent | 4f615adc1d7875f9fbe592619dc6b0f31cc7fd9e (diff) |
Add boolean type to tflite in favor of comparison implementations.
PiperOrigin-RevId: 192711203
Diffstat (limited to 'tensorflow/contrib/lite/interpreter.cc')
-rw-r--r-- | tensorflow/contrib/lite/interpreter.cc | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/interpreter.cc b/tensorflow/contrib/lite/interpreter.cc index 4575fe884d..f258654608 100644 --- a/tensorflow/contrib/lite/interpreter.cc +++ b/tensorflow/contrib/lite/interpreter.cc @@ -337,9 +337,13 @@ TfLiteStatus Interpreter::BytesRequired(TfLiteType type, const int* dims, case kTfLiteInt64: *bytes = sizeof(int64_t) * count; break; + case kTfLiteBool: + *bytes = sizeof(bool) * count; + break; default: - ReportError(&context_, - "Only float32, int32, int64, uint8 supported currently."); + ReportError( + &context_, + "Only float32, int32, int64, uint8, bool supported currently."); return kTfLiteError; } return kTfLiteOk; |