From 4320c92427ff97ef5433ef8b042f74e5110364df Mon Sep 17 00:00:00 2001
From: Teresa Charlin <teresa.charlinreyes@arm.com>
Date: Wed, 12 Aug 2020 16:04:41 +0100
Subject: 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 | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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