commit 4320c92427ff97ef5433ef8b042f74e5110364df
author Teresa Charlin <teresa.charlinreyes@arm.com> Wed Aug 12 16:04:41 2020 +0100
committer TeresaARM <teresa.charlinreyes@arm.com> Thu Aug 13 07:31:32 2020 +0000
tree 61e489bd84c33845bf303fed316ba53a83aeb049
parent 194b8ed3579450a3477182ed2691c86b1dd6a229

Revert "IVGCVSW-5209 Fix for SqueezeNet failing after change in Softmax"

This reverts commit fde170cc8cb7883ba7583a17ac3e8e07551bb6d5.

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Ieb542f49ad21ba13ddc10f0440ef08521c3f0451

src/armnnCaffeParser/CaffeParser.cpp

diff --git a/src/armnnCaffeParser/CaffeParser.cpp b/src/armnnCaffeParser/CaffeParser.cpp
index 8d44930..b8ce470 100644
--- a/src/armnnCaffeParser/CaffeParser.cpp
+++ b/src/armnnCaffeParser/CaffeParser.cpp
@@ -1219,7 +1219,7 @@
     //      Engine
 
     armnn::SoftmaxDescriptor softmaxDescriptor;
-    softmaxDescriptor.m_Axis = -1;
+    softmaxDescriptor.m_Axis = 1;
     armnn::IConnectableLayer* const softmaxLayer = m_Network->AddSoftmaxLayer(
         softmaxDescriptor,
         layerParam.name().c_str());