mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

fix cumprod ioname

This commit is contained in:
r1chardf1d0 2022-11-18 17:17:40 +08:00
parent 74e1f11b0d
commit a5dd4069ae
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/core/ops/cumprod.h
View File

@ -28,7 +28,7 @@ constexpr auto kNameCumProd = "CumProd";
class MIND_API CumProd : public BaseOperator { class MIND_API CumProd : public BaseOperator {
public: public:
MIND_API_BASE_MEMBER(CumProd); MIND_API_BASE_MEMBER(CumProd);
CumProd() : BaseOperator(kNameCumProd) {} CumProd() : BaseOperator(kNameCumProd) { InitIOName({"x", "axis"}, {"y"}); }
void Init(const bool exclusive, const bool reverse); void Init(const bool exclusive, const bool reverse);
void SetExclusive(const bool exclusive); void SetExclusive(const bool exclusive);
void SetReverse(const bool reverse); void SetReverse(const bool reverse);