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

add atan, atan2, expm1

This commit is contained in:
huangmengxi 2021-07-13 14:42:14 +08:00
parent 914f20abb9
commit 1bd448ef2a
3 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/_extends/graph_kernel/model/model.py
View File

@ -200,6 +200,9 @@ class PrimLib:
'CReal': Prim(ELEMWISE),
'CImag': Prim(ELEMWISE),
'Complex': Prim(ELEMWISE),
'Atan': Prim(ELEMWISE),
'Atan2': Prim(ELEMWISE),
'Expm1': Prim(ELEMWISE),
}
default_primtive = Prim(UNKNOWN)

3
mindspore/ccsrc/backend/optimizer/graph_kernel/graph_kernel_cluster.cc
View File

@ -62,6 +62,9 @@ std::vector<PrimitivePtr> GetClusterableOpList() {
prim::kPrimEqual,
prim::kPrimAssign,
prim::kPrimInplaceAssign,
prim::kPrimAtan,
prim::kPrimAtan2,
prim::kPrimExpm1,
#if ENABLE_D
prim::kPrimMatMul,
prim::KPrimTransData,

1
mindspore/core/base/core_ops.h
View File

@ -467,6 +467,7 @@ inline const PrimitivePtr kPrimAbs = std::make_shared<Primitive>("Abs");
inline const PrimitivePtr kPrimRint = std::make_shared<Primitive>("Rint");
inline const PrimitivePtr kPrimRound = std::make_shared<Primitive>("Round");
inline const PrimitivePtr kPrimExp = std::make_shared<Primitive>(kExp);
inline const PrimitivePtr kPrimExpm1 = std::make_shared<Primitive>("Expm1");
inline const PrimitivePtr kPrimLog = std::make_shared<Primitive>(kLog);
inline const PrimitivePtr kPrimRsqrt = std::make_shared<Primitive>("Rsqrt");
inline const PrimitivePtr kPrimRsqrtGrad = std::make_shared<Primitive>("RsqrtGrad");