diff options
author | Mike Kelly <mike.kelly@arm.com> | 2020-04-24 15:44:01 +0100 |
---|---|---|
committer | mike.kelly <mike.kelly@arm.com> | 2020-04-24 15:44:54 +0000 |
commit | ae42f01cb7f75fd5f7681f142f073f6f51e2e57f (patch) | |
tree | 9afb538ad8e2ae1e82c0a7152515b9610951522c /src | |
parent | ec52c3a04134e286d43cd2d22b916302f784ad29 (diff) | |
download | armnn-ae42f01cb7f75fd5f7681f142f073f6f51e2e57f.tar.gz |
Added new DataTypes to Serializer
* Added support for BFloat16 and QAsymmS8 to Serializr.
Signed-off-by: Mike Kelly <mike.kelly@arm.com>
Change-Id: I481af26d5758190ec2e5cd9f9d232d72bcb7acbd
Diffstat (limited to 'src')
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 10 |
1 files changed, 2 insertions, 8 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index cb7a5c456e..355673697f 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -1472,14 +1472,7 @@ flatbuffers::Offset<serializer::ConstTensor> break; } case armnn::DataType::Float16: - { - auto fbVector = CreateDataVector<int16_t>(constTensor.GetMemoryArea(), constTensor.GetNumBytes()); - flatbuffers::Offset<serializer::ShortData> flatBuffersData = serializer::CreateShortData( - m_flatBufferBuilder, - fbVector); - fbPayload = flatBuffersData.o; - break; - } + case armnn::DataType::BFloat16: case armnn::DataType::QSymmS16: { auto fbVector = CreateDataVector<int16_t>(constTensor.GetMemoryArea(), constTensor.GetNumBytes()); @@ -1490,6 +1483,7 @@ flatbuffers::Offset<serializer::ConstTensor> break; } case armnn::DataType::QSymmS8: + case armnn::DataType::QAsymmS8: case armnn::DataType::QAsymmU8: case armnn::DataType::Boolean: default: |