diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2020-03-27 15:56:57 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2020-03-27 15:56:57 +0000 |
commit | 1a84fe3b92aaab61d2f3b865d347459e16eb43d4 (patch) | |
tree | 140928df39ee8d46e3fb4e3ccc7883116a1ce929 /src/armnnSerializer/Serializer.hpp | |
parent | 57ef0088d20dd708ff92222d244ea02f1e1e5216 (diff) | |
download | armnn-1a84fe3b92aaab61d2f3b865d347459e16eb43d4.tar.gz |
IVGCVSW-4555 ArmnnConverter (Serializer) does not support per-axis quantization params
* TensorInfo can have multiple scales and quantization dimension.
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I0ff02e3766996b6a9da6dc4e92d366bc9505c77d
Diffstat (limited to 'src/armnnSerializer/Serializer.hpp')
-rw-r--r-- | src/armnnSerializer/Serializer.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/armnnSerializer/Serializer.hpp b/src/armnnSerializer/Serializer.hpp index 806caefc22..6dd655827d 100644 --- a/src/armnnSerializer/Serializer.hpp +++ b/src/armnnSerializer/Serializer.hpp @@ -293,6 +293,9 @@ private: flatbuffers::Offset<armnnSerializer::ConstTensor> CreateConstTensorInfo( const armnn::ConstTensor& constTensor); + /// Creates the serializer TensorInfo for the armnn TensorInfo. + flatbuffers::Offset<TensorInfo> CreateTensorInfo(const armnn::TensorInfo& tensorInfo); + template <typename T> flatbuffers::Offset<flatbuffers::Vector<T>> CreateDataVector(const void* memory, unsigned int size); |