diff options
Diffstat (limited to 'src/armnnTfLiteParser/TfLiteParser.cpp')
-rw-r--r-- | src/armnnTfLiteParser/TfLiteParser.cpp | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/src/armnnTfLiteParser/TfLiteParser.cpp b/src/armnnTfLiteParser/TfLiteParser.cpp index 86688add9d..1ee4950558 100644 --- a/src/armnnTfLiteParser/TfLiteParser.cpp +++ b/src/armnnTfLiteParser/TfLiteParser.cpp @@ -365,9 +365,15 @@ armnn::TensorInfo ToTensorInfo(TfLiteParser::TensorRawPtr tensorPtr, const std:: } } + std::vector<unsigned int> safeShape = shapes; + if (safeShape.size() == 0) + { + safeShape.push_back(1); + } + // two statements (on purpose) for easier debugging: - armnn::TensorInfo result(static_cast<unsigned int>(shapes.size()), - shapes.data(), + armnn::TensorInfo result(static_cast<unsigned int>(safeShape.size()), + safeShape.data(), type, quantizationScale, quantizationOffset); |