提交 f584416a 编写于 作者: M Megvii Engine Team

fix(dnn/bn): revise the conditions for inplace flag

GitOrigin-RevId: 59a104bf6a0dd4989c8a24f2314b7232796be818
上级 a9b60fbf
......@@ -137,7 +137,7 @@ SymbolVarArray BatchNormForward::make(SymbolVar x,
cg::OperatorNodeBase::NodeProp*
BatchNormForward::do_make_node_prop() const {
auto ret = Super::do_make_node_prop();
if (need_stats()) {
if (need_stats() && m_force_inplace) {
ret->add_flag(NodeProp::Flag::FORCE_UPDATE_INPUT_VAR);
}
return ret;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册