diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-04-17 11:22:38 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-04-17 11:55:30 +0100 |
commit | 4818d465973d72979a7f6e783e1b55b320781710 (patch) | |
tree | 7c6cce52a728b6fe3ea2733b808555ff27edbab2 /src/armnnTfLiteParser/TfLiteParser.cpp | |
parent | 0790dcea1056298d63f97dec904c8ade5d21f439 (diff) | |
download | armnn-4818d465973d72979a7f6e783e1b55b320781710.tar.gz |
IVGCVSW-2849 Add TfLite Parser support for Rank-0 operands and unit tests
Change-Id: I6dab12aed395a30466d66421c6e5a12659fedac8
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
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); |