This commit uses correct scale for Softmax likes #1728.
Signed-off-by: sjsujinkim <sjsujin.kim@samsung.com>
{
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
{
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;