提交 4dd2e472 编写于 作者: J Juncheng

fix



Former-commit-id: adbcd5892337be272571220858483ca9c3a97bee
上级 89aa2bc5
......@@ -55,7 +55,6 @@ void MultipleGatherOp::InferOutputBlobModelSplitAxis(
CHECK_EQ(parallel_context->policy(), ParallelPolicy::kDataParallel);
*ModelSplitAxis4BnInOp("out") = -1;
}
NaiveInferOutputBlobModelSplitAxis(ModelSplitAxis4BnInOp, ShapeNumAxes4BnInOp, parallel_context);
}
REGISTER_OP(OperatorConf::kMultipleGatherConf, MultipleGatherOp);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册