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

!45987 add infer map of ops 

Merge pull request !45987 from jjfeing/dynamic_static_op_bug_fix
This commit is contained in:
i-robot 2022-11-25 03:09:57 +00:00 committed by Gitee
parent 7831e21c36 bf7993594b
commit 142400c39a
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/core/abstract/ops/primitive_infer_map.cc
View File

@ -90,6 +90,7 @@
#include "ops/split.h"
#include "ops/grad/gelu_grad.h"
#include "ops/iou.h"
#include "ops/sparse_apply_proximal_adagrad.h"
namespace mindspore {
namespace abstract {
@ -442,6 +443,7 @@ PrimitiveEvalImplMap &GetPrimitiveToBackendEvalImplMap() {
{prim::kPrimConcatD, R{InferImplConcat, nullptr, true}},
{prim::kPrimPack, R{ops::StackInfer, nullptr, true}},
{prim::kPrimApplyAdamD, R{ops::ApplyAdamDInfer, nullptr, true}},
{prim::kPrimSparseApplyProximalAdagradD, R{ops::SparseApplyProximalAdagradInfer, nullptr, true}},
{prim::kPrimMul, R{ops::MulInfer, nullptr, true}},
{prim::kPrimMod, R{ops::ModInfer, nullptr, true}},
{prim::kPrimAdd, R{ops::AddInfer, nullptr, false}},

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

@ -989,6 +989,7 @@ GVAR_DEF(PrimitivePtr, kPrimApplyRMSProp, std::make_shared<Primitive>(kApplyRMSP
GVAR_DEF(PrimitivePtr, kPrimSparseApplyFtrl, std::make_shared<Primitive>("SparseApplyFtrl"));
GVAR_DEF(PrimitivePtr, kPrimSparseApplyFtrlD, std::make_shared<Primitive>("SparseApplyFtrlD"));
GVAR_DEF(PrimitivePtr, kPrimSparseApplyProximalAdagrad, std::make_shared<Primitive>("SparseApplyProximalAdagrad"));
GVAR_DEF(PrimitivePtr, kPrimSparseApplyProximalAdagradD, std::make_shared<Primitive>("SparseApplyProximalAdagradD"));
GVAR_DEF(PrimitivePtr, kPrimFusedAdam, std::make_shared<Primitive>("FusedAdam"));
GVAR_DEF(PrimitivePtr, kPrimFusedAdaFactor, std::make_shared<Primitive>("FusedAdaFactor"));
GVAR_DEF(PrimitivePtr, kPrimFusedAdaFactorWithGlobalNorm, std::make_shared<Primitive>("FusedAdaFactorWithGlobalNorm"));