From 6439dce107ebc1c2d983264c63452c9e27dfe62b Mon Sep 17 00:00:00 2001
From: yejianwu <yejianwu@xiaomi.com>
Date: Thu, 30 Aug 2018 20:46:46 +0800
Subject: [PATCH] update para 'value' to 'scalar_input' in eltwise

---
 mace/ops/lstmcell.h       | 2 +-
 mace/ops/lstmcell_test.cc | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/mace/ops/lstmcell.h b/mace/ops/lstmcell.h
index a4032379..c632cd3d 100644
--- a/mace/ops/lstmcell.h
+++ b/mace/ops/lstmcell.h
@@ -29,7 +29,7 @@ class LSTMCellOp : public Operator<D, T> {
   LSTMCellOp(const OperatorDef &op_def, Workspace *ws)
       : Operator<D, T>(op_def, ws),
         functor_(static_cast<T>(
-              OperatorBase::GetOptionalArg<float>("value", 0.0))) {}
+              OperatorBase::GetOptionalArg<float>("scalar_input", 0.0))) {}
 
   MaceStatus Run(StatsFuture *future) override {
     const Tensor *input = this->Input(INPUT);
diff --git a/mace/ops/lstmcell_test.cc b/mace/ops/lstmcell_test.cc
index 41866193..109096c5 100644
--- a/mace/ops/lstmcell_test.cc
+++ b/mace/ops/lstmcell_test.cc
@@ -84,7 +84,7 @@ void LSTMCellCPU(OpsTestNet *net,
 
   OpDefBuilder("Eltwise", "ForgetAdd")
       .Input("SplitOutput2")
-      .AddFloatArg("value", forget_add_name)
+      .AddFloatArg("scalar_input", forget_add_name)
       .AddIntArg("T", DataTypeToEnum<T>::v())
       .AddIntArg("type", static_cast<int>(kernels::EltwiseType::SUM))
       .Output("ForgetAdd")
@@ -176,7 +176,7 @@ void TestLSTMCell(const uint32_t &batch,
       .Input("WeightImage")
       .Input("BiasImage")
       .Input("PreCellImage")
-      .AddFloatArg("forget_add", forget_add)
+      .AddFloatArg("scalar_input", forget_add)
       .Output("CellImage")
       .Output("OutputImage")
       .Finalize(net.NewOperatorDef());
-- 
GitLab