diff options
| author | 박세희/On-Device Lab(SR)/Principal Engineer/삼성전자 <saehie.park@samsung.com> | 2020-03-31 09:39:54 +0900 |
|---|---|---|
| committer | GitHub Enterprise <noreply-CODE@samsung.com> | 2020-03-31 09:39:54 +0900 |
| commit | a6de0e7d3de46370d75cebf0fcfa36d0bec0ec64 (patch) | |
| tree | a91c6e044225cf0dadde084e28376884ebf0f987 | |
| parent | 86e050dcadb622eeb03bebd6082a687267a27bc4 (diff) | |
| download | nnfw-a6de0e7d3de46370d75cebf0fcfa36d0bec0ec64.tar.gz nnfw-a6de0e7d3de46370d75cebf0fcfa36d0bec0ec64.tar.bz2 nnfw-a6de0e7d3de46370d75cebf0fcfa36d0bec0ec64.zip | |
[tflchef] INT64 of tflite to tflchef type (#10767)
This will enable INT64 type conversion from tflite to tflchef
Signed-off-by: SaeHie Park <saehie.park@samsung.com>
| -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 111510560..ffdec1607 100644 --- a/compiler/tflchef/tflite/src/Convert.cpp +++ b/compiler/tflchef/tflite/src/Convert.cpp @@ -27,11 +27,12 @@ tflchef::TensorType as_tflchef_type(const tflite::TensorType type) return tflchef::FLOAT32; case tflite::TensorType_INT32: return tflchef::INT32; + case tflite::TensorType_INT64: + return tflchef::INT64; case tflite::TensorType_UINT8: return tflchef::UINT8; // TODO handle other types // TensorType_FLOAT16 - // TensorType_INT64 // TensorType_STRING // TensorType_BOOL // TensorType_INT16 |