diff options
| author | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-01-25 13:20:39 +0000 |
|---|---|---|
| committer | Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> | 2019-01-25 13:20:39 +0000 |
| commit | 5e9d29802e2cfbb13adc49c2a0ac9ba952dc7650 (patch) | |
| tree | 31baaf01ff767159fb6a4e594405b4acf03a6f51 /src/armnnUtils/ParserHelper.hpp | |
| parent | 6e2f60674cbe77c2a1da94ab71e35c298a1924de (diff) | |
| download | armnn-5e9d29802e2cfbb13adc49c2a0ac9ba952dc7650.tar.gz | |
IVGCVSW-2563 Fix bug in TfLiteParser::ParseConcatenation
Change-Id: I8fbf27b383a821e062f72809cc2e269fcd18851c
Diffstat (limited to 'src/armnnUtils/ParserHelper.hpp')
| -rw-r--r-- | src/armnnUtils/ParserHelper.hpp | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/src/armnnUtils/ParserHelper.hpp b/src/armnnUtils/ParserHelper.hpp index 24369dc521..bcc1e5b2cc 100644 --- a/src/armnnUtils/ParserHelper.hpp +++ b/src/armnnUtils/ParserHelper.hpp @@ -10,9 +10,11 @@ namespace armnnUtils { -void ProcessConcatInputTensorInfo(armnn::TensorInfo& inputTensorInfo, armnn::OriginsDescriptor& concatDescriptor, - const unsigned int& concatAxis, unsigned int inputIndex, - std::vector<unsigned int>& mergeDimSizes, unsigned int& mergeDim); +void ProcessConcatInputTensorInfo(armnn::TensorInfo& inputTensorInfo, + armnn::OriginsDescriptor& concatDescriptor, + const unsigned int& concatAxis, + unsigned int inputIndex, + unsigned int& mergeDimOrigin); /// Creates a tensor info after reducing the dimensions mentioned in axisData. void CalculateReducedOutputTensoInfo(const armnn::TensorInfo& inputTensorInfo, const armnn::TensorInfo& axisTensorInfo, |