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

add infer map of SparseApplyProximalAdagradInfer

This commit is contained in:
jjfeing 2022-11-24 19:42:42 +08:00
parent 98d9e95256
commit bf7993594b
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

@ -988,6 +988,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"));