From 98c59e70f8021bb1f8669fb0f61d48d1f20cfa14 Mon Sep 17 00:00:00 2001
From: "Efimov Alexander/AI Tools Lab/./Samsung Electronics"
 <a.efimov@samsung.com>
Date: Fri, 17 Aug 2018 10:38:21 +0300
Subject: [PATCH] Fix softmax axis inporting in TF lite (#1038)

Set softmax axis to 1 in case of fully connected layer before it

Signed-off-by: Efimov Alexander <a.efimov@samsung.com>
---
 contrib/nnc/libs/frontend/tflite/src/tflite_op_creator.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/contrib/nnc/libs/frontend/tflite/src/tflite_op_creator.cpp b/contrib/nnc/libs/frontend/tflite/src/tflite_op_creator.cpp
index 629aa9a..2f7e84e 100644
--- a/contrib/nnc/libs/frontend/tflite/src/tflite_op_creator.cpp
+++ b/contrib/nnc/libs/frontend/tflite/src/tflite_op_creator.cpp
@@ -77,7 +77,7 @@ std::vector<INode::Ref> OpCreator::createSoftmax(InputOps inputs, InputParams pa
                                                  const SoftmaxOptions *opts)
 {
   // TODO: here assuming that softmax is applied to a 1-d tensor
-  return createOp<ops::SoftmaxOp>(inputs, ActivationFunctionType_NONE, 0);
+  return createOp<ops::SoftmaxOp>(inputs, ActivationFunctionType_NONE, 1);
 }
 
 std::vector<INode::Ref> OpCreator::createReshape(InputOps inputs, InputParams params,
-- 
2.7.4