[neurun] Use correct scale for Softmax (#2433)

This commit uses correct scale for Softmax likes #1728.

Signed-off-by: sjsujinkim <sjsujin.kim@samsung.com>

diff --git a/runtimes/neurun/src/backend/acl_cl/StageGenerator.cc b/runtimes/neurun/src/backend/acl_cl/StageGenerator.cc
index 723c32e..bb76009 100644 (file)
--- a/runtimes/neurun/src/backend/acl_cl/StageGenerator.cc
+++ b/runtimes/neurun/src/backend/acl_cl/StageGenerator.cc
@@ -481,6 +481,9 @@ Stage StageGenerator::generate(const ::internal::tflite::op::Softmax::Node &node
 {
   const ::neurun::graph::operand::Index output_index{node.param().output_index};
   const ::neurun::graph::operand::Index input_index{node.param().input_index};
+  const ::neurun::graph::operand::Index scale_index{node.param().scale_index};
+
+  assert(_ctx.at(scale_index).shape().rank() == 0);
 
   struct Param
   {
@@ -493,8 +496,7 @@ Stage StageGenerator::generate(const ::internal::tflite::op::Softmax::Node &node
 
   param.output_index = output_index.asInt();
   param.input_index = input_index.asInt();
-  // TODO Set scale correctly
-  param.scale = 1.0f;
+  param.scale = _ctx.at(scale_index).asScalar<float>();
 
   auto tensors = _tensor_builder;