提交 980a6753 编写于 作者: T Tao Luo

fix typo to pass the ci

test=develop
上级 e0d4e04b
...@@ -29,6 +29,7 @@ void SetOp(ProgramDesc* prog, const std::string& type, ...@@ -29,6 +29,7 @@ void SetOp(ProgramDesc* prog, const std::string& type,
if (type == "mul") { if (type == "mul") {
op->SetInput("X", {inputs[0]}); op->SetInput("X", {inputs[0]});
op->SetInput("Y", {inputs[1]}); op->SetInput("Y", {inputs[1]});
op->SetAttr("x_num_col_dims", {1});
} else if (type == "elementwise_add") { } else if (type == "elementwise_add") {
op->SetInput("X", inputs); op->SetInput("X", inputs);
} }
......
...@@ -121,7 +121,7 @@ void FCOpMaker::Make() { ...@@ -121,7 +121,7 @@ void FCOpMaker::Make() {
AddInput("W", "(Tensor), The weight fc op with shape (I, O)."); AddInput("W", "(Tensor), The weight fc op with shape (I, O).");
AddInput("Bias", "(Tensor, optional) Bias vector with shape (1 x O") AddInput("Bias", "(Tensor, optional) Bias vector with shape (1 x O")
.AsDispensable(); .AsDispensable();
AddAttr<int>("x_num_col_dims", AddAttr<int>("in_num_col_dims",
"(int, default 1), The fc op can take tensors with more than " "(int, default 1), The fc op can take tensors with more than "
"two dimensions as its inputs.") "two dimensions as its inputs.")
.SetDefault(1) .SetDefault(1)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册