未验证 提交 c57ab97f 编写于 作者: H hong 提交者: GitHub

[NewIR]Fix new ir bn grad bug (#56347)

* fix op translator reshape type

* new ir support vector type place transfer

* add test case

* fix new ir bn grad bug
上级 fd765f61
......@@ -444,6 +444,9 @@ void BatchNormGradMaker<T>::Apply(GradOpPtr<T> op) const {
op->SetInput("Variance", this->Output("VarianceOut"));
}
op->SetInput("MeanOut", this->Output("MeanOut"));
op->SetInput("VarianceOut", this->Output("VarianceOut"));
op->SetAttrMap(this->Attrs());
op->SetOutput(framework::GradVarName("X"), this->InputGrad("X"));
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册