From 8ac5fc39b2556fd3ac0d0c43ebf55525bde97f58 Mon Sep 17 00:00:00 2001
From: Kevin May <kevin.may@arm.com>
Date: Thu, 12 Dec 2019 17:28:05 +0000
Subject: IVGCVSW-4262 Use ACL Permute and Reshape Validate function in Neon
 and CL

!android-nn-driver:2465

Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: Ibabb73c0ae0df2e530a68398f75c76e6b80c0701
---
 src/backends/backendsCommon/WorkloadFactory.cpp | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'src/backends/backendsCommon/WorkloadFactory.cpp')

diff --git a/src/backends/backendsCommon/WorkloadFactory.cpp b/src/backends/backendsCommon/WorkloadFactory.cpp
index ffef5b4eb7..cda822c3d7 100644
--- a/src/backends/backendsCommon/WorkloadFactory.cpp
+++ b/src/backends/backendsCommon/WorkloadFactory.cpp
@@ -712,7 +712,9 @@ bool IWorkloadFactory::IsLayerSupported(const BackendId& backendId,
         {
             auto cLayer = boost::polymorphic_downcast<const ReshapeLayer*>(&layer);
             const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo();
+            const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo();
             result = layerSupportObject->IsReshapeSupported(OverrideDataType(input, dataType),
+                                                            OverrideDataType(output, dataType),
                                                             cLayer->GetParameters(),
                                                             reason);
             break;
-- 
cgit v1.2.1