aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/LayerSupport.cpp
diff options
context:
space:
mode:
authorKevin May <kevin.may@arm.com>2019-12-12 17:28:05 +0000
committerTeresaARM <teresa.charlinreyes@arm.com>2020-01-24 16:47:17 +0000
commita023c40a7644f0b32f8f593b9a3614c92b5f933d (patch)
tree996a35023c6ea6adaee76472e49857fd545ad148 /src/armnn/LayerSupport.cpp
parenta0d189697bb549a37886790536be7e6879ee943a (diff)
downloadarmnn-a023c40a7644f0b32f8f593b9a3614c92b5f933d.tar.gz
IVGCVSW-4262 Use ACL Permute and Reshape Validate function in Neon and CL
!android-nn-driver:2642 Signed-off-by: Kevin May <kevin.may@arm.com> Change-Id: Ibabb73c0ae0df2e530a68398f75c76e6b80c0701
Diffstat (limited to 'src/armnn/LayerSupport.cpp')
-rw-r--r--src/armnn/LayerSupport.cpp3
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/LayerSupport.cpp b/src/armnn/LayerSupport.cpp
index 08d91fc20b..3c244b0454 100644
--- a/src/armnn/LayerSupport.cpp
+++ b/src/armnn/LayerSupport.cpp
@@ -530,11 +530,12 @@ bool IsPreluSupported(const BackendId& backend,
bool IsReshapeSupported(const BackendId& backend,
const TensorInfo& input,
+ const TensorInfo& output,
const ReshapeDescriptor& descriptor,
char* reasonIfUnsupported,
size_t reasonIfUnsupportedMaxLength)
{
- FORWARD_LAYER_SUPPORT_FUNC(backend, IsReshapeSupported, input, descriptor);
+ FORWARD_LAYER_SUPPORT_FUNC(backend, IsReshapeSupported, input, output, descriptor);
}
bool IsResizeSupported(const BackendId& backend,