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

fix cumsum prim op maker type error (#51014)

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