diff options
author | Jerry Ge <jerry.ge@arm.com> | 2023-05-08 19:17:22 +0000 |
---|---|---|
committer | Jerry Ge <jerry.ge@arm.com> | 2023-05-08 19:42:24 +0000 |
commit | b413a957b5a1bd4270a6d1270ad9a705d96864c8 (patch) | |
tree | 2213f761935c7445dd91fe56c1e726c2802076fd | |
parent | cfcb20d08c4c409bbcd2d2dde6ca5ecdac299454 (diff) | |
download | serialization_lib-b413a957b5a1bd4270a6d1270ad9a705d96864c8.tar.gz |
Fix empty shaped serialized tensor
Signed-off-by: Jerry Ge <jerry.ge@arm.com>
Change-Id: I512181cca1a3b1ed5d9a5435dbf11d7efcdbc2a5
-rw-r--r-- | src/tosa_serialization_handler.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/tosa_serialization_handler.cpp b/src/tosa_serialization_handler.cpp index d782213..25ac5df 100644 --- a/src/tosa_serialization_handler.cpp +++ b/src/tosa_serialization_handler.cpp @@ -26,7 +26,10 @@ TosaSerializationTensor::TosaSerializationTensor(const flatbuffers::String* name { _dtype = dtype; - std::copy(shape->begin(), shape->end(), std::back_inserter(_shape)); + if (shape) + { + std::copy(shape->begin(), shape->end(), std::back_inserter(_shape)); + } assert(name); _name = name->str(); |