From b413a957b5a1bd4270a6d1270ad9a705d96864c8 Mon Sep 17 00:00:00 2001 From: Jerry Ge Date: Mon, 8 May 2023 19:17:22 +0000 Subject: Fix empty shaped serialized tensor Signed-off-by: Jerry Ge Change-Id: I512181cca1a3b1ed5d9a5435dbf11d7efcdbc2a5 --- src/tosa_serialization_handler.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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(); -- cgit v1.2.1