From 3b9383548309a8f9121668826a628ec250c57a1c Mon Sep 17 00:00:00 2001
From: Francis Murtagh <francis.murtagh@arm.com>
Date: Fri, 26 Jul 2019 15:44:17 +0100
Subject: IVGCVSW-3554 Update workloads to pass Softmax Axis Parameter to
 Backends

 * Add check in CL and Neon to ensure axis is 1 otherwise
   return unsupported.
 * Edit CreateWorkload test and JsonPrinter test to ensure axis of 1.

Change-Id: I499b405532e26fefc2dd1c18b6dc6005813b5604
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
---
 src/armnnCaffeParser/CaffeParser.cpp | 1 +
 1 file changed, 1 insertion(+)

(limited to 'src/armnnCaffeParser')

diff --git a/src/armnnCaffeParser/CaffeParser.cpp b/src/armnnCaffeParser/CaffeParser.cpp
index 4027e9481b..c73365bae1 100644
--- a/src/armnnCaffeParser/CaffeParser.cpp
+++ b/src/armnnCaffeParser/CaffeParser.cpp
@@ -1220,6 +1220,7 @@ void CaffeParserBase::ParseSoftmaxLayer(const LayerParameter& layerParam)
     //      Engine
 
     armnn::SoftmaxDescriptor softmaxDescriptor;
+    softmaxDescriptor.m_Axis = 1;
     armnn::IConnectableLayer* const softmaxLayer = m_Network->AddSoftmaxLayer(
         softmaxDescriptor,
         layerParam.name().c_str());
-- 
cgit v1.2.1