提交 6439dce1 编写于 作者: Y yejianwu

update para 'value' to 'scalar_input' in eltwise

上级 f9ad189c
......@@ -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);
......
......@@ -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());
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册