aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/SerializerUtils.cpp
diff options
context:
space:
mode:
authorMike Kelly <mike.kelly@arm.com>2019-02-19 17:22:07 +0000
committermike.kelly <mike.kelly@arm.com>2019-02-19 18:21:11 +0000
commita0766c3d461f6a635ef0c41c83c3ba868f2fd21a (patch)
tree3565ace677fd6de80dc30476c00face7e885d46a /src/armnnSerializer/SerializerUtils.cpp
parent2b183fb359774cbac5d628579ec2b4a7b6b41def (diff)
downloadarmnn-a0766c3d461f6a635ef0c41c83c3ba868f2fd21a.tar.gz
IVGCVSW-2646 Add Serializer & Deserializer for Conv2D
* Added Convolution2dLayer to Schema.fbs * Added ConstTensorData serialization and deserialization helper functions * Added Convolution2d serialization and deserialization support * Added serialization and deserialization unit tests Change-Id: Id376c08410ae01511972a2b0abdce9cfab907462 Signed-off-by: Mike Kelly <mike.kelly@arm.com> Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Diffstat (limited to 'src/armnnSerializer/SerializerUtils.cpp')
-rw-r--r--src/armnnSerializer/SerializerUtils.cpp17
1 files changed, 17 insertions, 0 deletions
diff --git a/src/armnnSerializer/SerializerUtils.cpp b/src/armnnSerializer/SerializerUtils.cpp
index 5772eab56c..2bad85e1a0 100644
--- a/src/armnnSerializer/SerializerUtils.cpp
+++ b/src/armnnSerializer/SerializerUtils.cpp
@@ -11,6 +11,23 @@ namespace armnnSerializer
using namespace armnn;
namespace serializer = armnn::armnnSerializer;
+serializer::ConstTensorData GetFlatBufferConstTensorData(DataType dataType)
+{
+ switch (dataType)
+ {
+ case DataType::Float32:
+ case DataType::Signed32:
+ return serializer::ConstTensorData::ConstTensorData_IntData;
+ case DataType::Float16:
+ return serializer::ConstTensorData::ConstTensorData_ShortData;
+ case DataType::QuantisedAsymm8:
+ case DataType::Boolean:
+ return serializer::ConstTensorData::ConstTensorData_ByteData;
+ default:
+ return serializer::ConstTensorData::ConstTensorData_NONE;
+ }
+}
+
serializer::DataType GetFlatBufferDataType(DataType dataType)
{
switch (dataType)