Add Signed64 tensor data type support
authorInki Dae <inki.dae@samsung.com>
Mon, 3 Aug 2020 09:36:32 +0000 (18:36 +0900)
committerInki Dae <inki.dae@samsung.com>
Mon, 3 Aug 2020 09:36:32 +0000 (18:36 +0900)
Change-Id: I30d70142185158d1eee5f2031e02828124dfd9ce
Signed-off-by: Inki Dae <inki.dae@samsung.com>
src/inference_engine_armnn.cpp

index b041db80be7fb6d8adca36c9565ef3b4a10bd8d5..52ad3a5d5b9e1bffba38f8b3800856b6738b0905 100644 (file)
@@ -77,6 +77,9 @@ namespace ARMNNImpl
                case armnn::DataType::QuantisedAsymm8:
                        data_type = INFERENCE_TENSOR_DATA_TYPE_UINT8;
                        break;
+               case armnn::DataType::Signed64:
+                       data_type = INFERENCE_TENSOR_DATA_TYPE_UINT64;
+                       break;
                default:
                        LOGE("Invalid Input tensor type so it will use float32 in default.");
                        data_type = INFERENCE_TENSOR_DATA_TYPE_FLOAT32;