From fde170cc8cb7883ba7583a17ac3e8e07551bb6d5 Mon Sep 17 00:00:00 2001
From: Teresa Charlin <teresa.charlinreyes@arm.com>
Date: Wed, 12 Aug 2020 11:46:43 +0100
Subject: IVGCVSW-5209 Fix for SqueezeNet failing after change in Softmax

Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: Iaf298384a9010715ea7040460cbc537d2cb0ab64
---
 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 b8ce470505..8d44930d59 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