diff options
Diffstat (limited to 'compiler/tflchef/tflite/src/Convert.cpp')
-rw-r--r-- | compiler/tflchef/tflite/src/Convert.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/compiler/tflchef/tflite/src/Convert.cpp b/compiler/tflchef/tflite/src/Convert.cpp index f47e51d3d..242987661 100644 --- a/compiler/tflchef/tflite/src/Convert.cpp +++ b/compiler/tflchef/tflite/src/Convert.cpp @@ -35,8 +35,9 @@ tflchef::TensorType as_tflchef_type(const tflite::TensorType type) return tflchef::BOOL; case tflite::TensorType_INT16: return tflchef::INT16; + case tflite::TensorType_FLOAT16: + return tflchef::FLOAT16; // TODO handle other types - // TensorType_FLOAT16 // TensorType_STRING // TensorType_COMPLEX64 default: |