diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2023-06-14 14:51:17 +0100 |
---|---|---|
committer | TeresaARM <teresa.charlinreyes@arm.com> | 2023-06-22 12:12:31 +0000 |
commit | a7a605a7554158a47127b1988b94c6bfac0a64b9 (patch) | |
tree | 9dce948707b266e45b0a5a6974ca940b54ce7f42 /src/armnnTestUtils/GraphUtils.cpp | |
parent | a9c5c16154386fb1ec924c3778caf6b027bb57ee (diff) | |
download | armnn-a7a605a7554158a47127b1988b94c6bfac0a64b9.tar.gz |
IVGCVSW-7785 BugFix: ExpandDims not working when batch!=1
* This commit fixes partially the ticket.
In ToTensorInfo() we assume batch is 1 when it is unknown.
We call OutputTensorInfoFromInputs() to amend this assumption/
However, this does not work for reshape layer.
Therefore, we have to calculate the output shape in the ParseExpandDims().
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Iedc32a44b4ec0d8b7d2cc0b08f38f0776402f7bd
Diffstat (limited to 'src/armnnTestUtils/GraphUtils.cpp')
0 files changed, 0 insertions, 0 deletions