aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/SerializerUtils.cpp
diff options
context:
space:
mode:
authorMike Kelly <mike.kelly@arm.com>2021-04-06 12:25:55 +0100
committerMatthew Sloyan <matthew.sloyan@arm.com>2021-04-09 12:38:48 +0100
commit1f140f7226c4ed7bc5cbaf2ce09654eee452f4bf (patch)
tree044d491d429b6da4d85530afd4ea2a310cdbb827 /src/armnnSerializer/SerializerUtils.cpp
parent7c67fabc86b6647855beebac9f6cfe92341357cb (diff)
downloadarmnn-1f140f7226c4ed7bc5cbaf2ce09654eee452f4bf.tar.gz
MLCE-328 Serializer/Deserializer does not support Signed64
* Added support for Signed64 to flatbuffer's schema & updated source tree * Added support for Signed64 to TFLite Delegate * Added support for Signed64 to Serializer * Added support for Signed64 to Deserializer * Added unit test for ArgMinMax to Deserializer * Deprecated m_Output_Type from the ArgMinMaxDescriptor: the output type is solely determined by the DataType of the output Tensor * Fixed issue where RefArgMinMaxWorkload could output data using the wrong DataType * Added Signed64 to RefLayerSupport::IsArgMinMaxSupported as a supported type Signed-off-by: Mike Kelly <mike.kelly@arm.com> Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com> Change-Id: Ib622c052a1f8aa3e658262f8bde5a6881a8cbe10
Diffstat (limited to 'src/armnnSerializer/SerializerUtils.cpp')
-rw-r--r--src/armnnSerializer/SerializerUtils.cpp6
1 files changed, 5 insertions, 1 deletions
diff --git a/src/armnnSerializer/SerializerUtils.cpp b/src/armnnSerializer/SerializerUtils.cpp
index 929bf92d24..936fb531d5 100644
--- a/src/armnnSerializer/SerializerUtils.cpp
+++ b/src/armnnSerializer/SerializerUtils.cpp
@@ -1,5 +1,5 @@
//
-// Copyright © 2017 Arm Ltd. All rights reserved.
+// Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
@@ -57,6 +57,8 @@ armnnSerializer::ConstTensorData GetFlatBufferConstTensorData(armnn::DataType da
case armnn::DataType::QSymmS8:
case armnn::DataType::Boolean:
return armnnSerializer::ConstTensorData::ConstTensorData_ByteData;
+ case armnn::DataType::Signed64:
+ return armnnSerializer::ConstTensorData::ConstTensorData_LongData;
default:
return armnnSerializer::ConstTensorData::ConstTensorData_NONE;
}
@@ -72,6 +74,8 @@ armnnSerializer::DataType GetFlatBufferDataType(armnn::DataType dataType)
return armnnSerializer::DataType::DataType_Float16;
case armnn::DataType::Signed32:
return armnnSerializer::DataType::DataType_Signed32;
+ case armnn::DataType::Signed64:
+ return armnnSerializer::DataType::DataType_Signed64;
case armnn::DataType::QSymmS16:
return armnnSerializer::DataType::DataType_QSymmS16;
case armnn::DataType::QAsymmS8: