未验证 提交 add510b9 编写于 作者: G GGBond8488 提交者: GitHub

fix cumsum prim op maker type error (#51014)

上级 c9309942
...@@ -117,7 +117,7 @@ class CumsumCompositeGradOpMaker : public prim::CompositeGradOpMakerBase { ...@@ -117,7 +117,7 @@ class CumsumCompositeGradOpMaker : public prim::CompositeGradOpMakerBase {
bool flatten = static_cast<bool>(this->Attr<bool>("flatten")); bool flatten = static_cast<bool>(this->Attr<bool>("flatten"));
bool exclusive = static_cast<bool>(this->Attr<bool>("exclusive")); bool exclusive = static_cast<bool>(this->Attr<bool>("exclusive"));
bool reverse = static_cast<bool>(this->Attr<bool>("reverse")); bool reverse = static_cast<bool>(this->Attr<bool>("reverse"));
VLOG(6) << "Runing add_grad composite func"; VLOG(6) << "Runing cumsum composite func";
prim::cumsum_grad<prim::DescTensor>( prim::cumsum_grad<prim::DescTensor>(
x, out_grad, axis, flatten, exclusive, reverse, dx_ptr); x, out_grad, axis, flatten, exclusive, reverse, dx_ptr);
this->RecoverOutputName(dx, dx_name); this->RecoverOutputName(dx, dx_name);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册