Browse Source

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

GitOrigin-RevId: 59a104bf6a
release-1.5
Megvii Engine Team 4 years ago
parent
commit
f584416aa2
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      src/opr/impl/dnn/batch_norm.cpp

+ 1
- 1
src/opr/impl/dnn/batch_norm.cpp View File

@@ -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;


Loading…
Cancel
Save