diff options
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); |