aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorKevin May <kevin.may@arm.com>2019-02-22 09:40:55 +0000
committerKevin May <kevin.may@arm.com>2019-02-22 09:40:55 +0000
commit843b61ab0bb8200818e356f253b86da03bc74ba8 (patch)
tree0f3a25910ffc4933f94573fb591d9666d4fee9f1
parent216ce86bb4a293b50737ee6560b2728a858b27d2 (diff)
downloadandroid-nn-driver-843b61ab0bb8200818e356f253b86da03bc74ba8.tar.gz
MLCE-87 Fix ConvertFullyConnected not supported
Change-Id: If9f06b5b117972a43b5d1516c27ad350fafc4263 Signed-off-by: Kevin May <kevin.may@arm.com>
-rw-r--r--1.0/HalPolicy.cpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/1.0/HalPolicy.cpp b/1.0/HalPolicy.cpp
index d48b5494..34fdd279 100644
--- a/1.0/HalPolicy.cpp
+++ b/1.0/HalPolicy.cpp
@@ -670,7 +670,7 @@ bool HalPolicy::ConvertFullyConnected(const Operation& operation, const Model& m
if (!IsLayerSupported(__func__,
armnn::IsFullyConnectedSupported,
data.m_Compute,
- inputInfo,
+ reshapedInfo,
outputInfo,
weights.GetInfo(),
bias.GetInfo(),