From 5f26583da4b7c2d04d98021fd05b828572782f7d Mon Sep 17 00:00:00 2001
From: yujianfeng <yujianfeng5@huawei.com>
Date: Mon, 11 May 2020 18:40:43 +0800
Subject: [PATCH] Add some op registrations for const to attr

---
 .../pass/const_input_to_attr_registry.cc           | 13 ++++++++++++-
 mindspore/ccsrc/utils/utils.h                      | 14 +++++++++++---
 2 files changed, 23 insertions(+), 4 deletions(-)

diff --git a/mindspore/ccsrc/pre_activate/pass/const_input_to_attr_registry.cc b/mindspore/ccsrc/pre_activate/pass/const_input_to_attr_registry.cc
index 3153a3bef..bb4a710d4 100644
--- a/mindspore/ccsrc/pre_activate/pass/const_input_to_attr_registry.cc
+++ b/mindspore/ccsrc/pre_activate/pass/const_input_to_attr_registry.cc
@@ -39,6 +39,7 @@ ConstInputToAttrInfoRegistry::ConstInputToAttrInfoRegistry() {
   Register(prim::kPrimTranspose->name(), {1});
   Register(prim::kPrimUnsortedSegmentSum->name(), {2});
   Register(prim::kPrimOneHot->name(), {1});
+  Register(prim::kPrimConcat->name(), {0});
   Register(kUnsortedSegmentProdOpName, {2});
   Register(kUnsortedSegmentMinOpName, {2});
   Register(kSimpleMeanGradOpName, {1});
@@ -55,7 +56,17 @@ ConstInputToAttrInfoRegistry::ConstInputToAttrInfoRegistry() {
   Register(kSplitOpName, {0});
   Register(kErfOpName, {1});
   Register(kSparseApplyAdagradOpName, {2});
-  Register(kResizeNearestNeighborGrad, {1});
+  Register(kResizeNearestNeighborGradOpName, {1});
+  Register(kResizeNearestNeighborV2OpName, {1});
+  Register(kResizeNearestNeighborV2GradOpName, {1});
+  Register(kApplyRMSPropOpname, {4, 5, 6});
+  Register(kCumsumOpName, {1});
+  Register(kResizeBilinearV2OpName, {1});
+  Register(kReduceProdOpName, {1});
+  Register(kCumprodOpName, {1});
+  Register(kSpaceToBatchOpName, {1});
+  Register(kBatchToSpaceOpName, {1});
+  Register(kPadOpName, {1});
 }
 
 ConstInputToAttrInfoRegistry &ConstInputToAttrInfoRegistry::Instance() {
diff --git a/mindspore/ccsrc/utils/utils.h b/mindspore/ccsrc/utils/utils.h
index b1d71c18a..d5313247d 100644
--- a/mindspore/ccsrc/utils/utils.h
+++ b/mindspore/ccsrc/utils/utils.h
@@ -107,11 +107,9 @@ constexpr auto kLayerNormBetaGammaBackpropOpName = "LayerNormBetaGammaBackprop";
 constexpr auto kLambNextMVOpName = "LambNextMV";
 constexpr auto kConfusionTransposeDOpName = "ConfusionTransposeD";
 constexpr auto kAdamApplyOneWithDecayOpName = "AdamApplyOneWithDecay";
-constexpr auto kBatchNormOpName = "BatchNorm";
 constexpr auto kBatchNormGradOpName = "BatchNormGrad";
 constexpr auto kAdamApplyOneOpName = "AdamApplyOne";
-constexpr auto kDropoutGenMask = "DropoutGenMask";
-constexpr auto kResizeNearestNeighborGrad = "ResizeNearestNeighborGrad";
+constexpr auto kResizeNearestNeighborGradOpName = "ResizeNearestNeighborGrad";
 constexpr auto kFusedMulAddOpName = "FusedMulAdd";
 constexpr auto kFusedMulAddNOpName = "FusedMulAddN";
 constexpr auto kFusedMulApplyMomentumOpName = "FusedMulApplyMomentum";
@@ -125,6 +123,16 @@ constexpr auto kRecvOpName = "Recv";
 constexpr auto kReluV2OpName = "ReLUV2";
 constexpr auto kReluGradV2OpName = "ReluGradV2";
 constexpr auto kAddNOpName = "AddN";
+constexpr auto kResizeNearestNeighborV2OpName = "ResizeNearestNeighborV2";
+constexpr auto kResizeNearestNeighborV2GradOpName = "ResizeNearestNeighborV2Grad";
+constexpr auto kApplyRMSPropOpname = "ApplyRMSProp";
+constexpr auto kCumsumOpName = "Cumsum";
+constexpr auto kResizeBilinearV2OpName = "kResizeBilinearV2";
+constexpr auto kReduceProdOpName = "ReduceProd";
+constexpr auto kCumprodOpName = "Cumprod";
+constexpr auto kSpaceToBatchOpName = "SpaceToBatch";
+constexpr auto kBatchToSpaceOpName = "BatchToSpace";
+constexpr auto kPadOpName = "Pad";
 constexpr auto kConv2DBackpropInputOpName = "Conv2DBackpropInput";
 constexpr auto kFusionOpConv2DBackpropInputReluGradV2Name = "FusionOp_Conv2DBackpropInput_ReluGradV2";
 constexpr auto kFusionOpConv2DBackpropInputAddNReluGradV2Name = "FusionOp_Conv2DBackpropInput_AddN_ReluGradV2";
-- 
GitLab