提交 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, ...@@ -137,7 +137,7 @@ SymbolVarArray BatchNormForward::make(SymbolVar x,
cg::OperatorNodeBase::NodeProp* cg::OperatorNodeBase::NodeProp*
BatchNormForward::do_make_node_prop() const { BatchNormForward::do_make_node_prop() const {
auto ret = Super::do_make_node_prop(); 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); ret->add_flag(NodeProp::Flag::FORCE_UPDATE_INPUT_VAR);
} }
return ret; return ret;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册