提交 c1290e7b 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!1088 Add some op registrations for const value to attr

Merge pull request !1088 from YuJianfeng/master
...@@ -39,6 +39,7 @@ ConstInputToAttrInfoRegistry::ConstInputToAttrInfoRegistry() { ...@@ -39,6 +39,7 @@ ConstInputToAttrInfoRegistry::ConstInputToAttrInfoRegistry() {
Register(prim::kPrimTranspose->name(), {1}); Register(prim::kPrimTranspose->name(), {1});
Register(prim::kPrimUnsortedSegmentSum->name(), {2}); Register(prim::kPrimUnsortedSegmentSum->name(), {2});
Register(prim::kPrimOneHot->name(), {1}); Register(prim::kPrimOneHot->name(), {1});
Register(prim::kPrimConcat->name(), {0});
Register(kUnsortedSegmentProdOpName, {2}); Register(kUnsortedSegmentProdOpName, {2});
Register(kUnsortedSegmentMinOpName, {2}); Register(kUnsortedSegmentMinOpName, {2});
Register(kSimpleMeanGradOpName, {1}); Register(kSimpleMeanGradOpName, {1});
...@@ -55,7 +56,17 @@ ConstInputToAttrInfoRegistry::ConstInputToAttrInfoRegistry() { ...@@ -55,7 +56,17 @@ ConstInputToAttrInfoRegistry::ConstInputToAttrInfoRegistry() {
Register(kSplitOpName, {0}); Register(kSplitOpName, {0});
Register(kErfOpName, {1}); Register(kErfOpName, {1});
Register(kSparseApplyAdagradOpName, {2}); 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() { ConstInputToAttrInfoRegistry &ConstInputToAttrInfoRegistry::Instance() {
......
...@@ -107,11 +107,9 @@ constexpr auto kLayerNormBetaGammaBackpropOpName = "LayerNormBetaGammaBackprop"; ...@@ -107,11 +107,9 @@ constexpr auto kLayerNormBetaGammaBackpropOpName = "LayerNormBetaGammaBackprop";
constexpr auto kLambNextMVOpName = "LambNextMV"; constexpr auto kLambNextMVOpName = "LambNextMV";
constexpr auto kConfusionTransposeDOpName = "ConfusionTransposeD"; constexpr auto kConfusionTransposeDOpName = "ConfusionTransposeD";
constexpr auto kAdamApplyOneWithDecayOpName = "AdamApplyOneWithDecay"; constexpr auto kAdamApplyOneWithDecayOpName = "AdamApplyOneWithDecay";
constexpr auto kBatchNormOpName = "BatchNorm";
constexpr auto kBatchNormGradOpName = "BatchNormGrad"; constexpr auto kBatchNormGradOpName = "BatchNormGrad";
constexpr auto kAdamApplyOneOpName = "AdamApplyOne"; constexpr auto kAdamApplyOneOpName = "AdamApplyOne";
constexpr auto kDropoutGenMask = "DropoutGenMask"; constexpr auto kResizeNearestNeighborGradOpName = "ResizeNearestNeighborGrad";
constexpr auto kResizeNearestNeighborGrad = "ResizeNearestNeighborGrad";
constexpr auto kFusedMulAddOpName = "FusedMulAdd"; constexpr auto kFusedMulAddOpName = "FusedMulAdd";
constexpr auto kFusedMulAddNOpName = "FusedMulAddN"; constexpr auto kFusedMulAddNOpName = "FusedMulAddN";
constexpr auto kFusedMulApplyMomentumOpName = "FusedMulApplyMomentum"; constexpr auto kFusedMulApplyMomentumOpName = "FusedMulApplyMomentum";
...@@ -125,6 +123,16 @@ constexpr auto kRecvOpName = "Recv"; ...@@ -125,6 +123,16 @@ constexpr auto kRecvOpName = "Recv";
constexpr auto kReluV2OpName = "ReLUV2"; constexpr auto kReluV2OpName = "ReLUV2";
constexpr auto kReluGradV2OpName = "ReluGradV2"; constexpr auto kReluGradV2OpName = "ReluGradV2";
constexpr auto kAddNOpName = "AddN"; 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 kConv2DBackpropInputOpName = "Conv2DBackpropInput";
constexpr auto kFusionOpConv2DBackpropInputReluGradV2Name = "FusionOp_Conv2DBackpropInput_ReluGradV2"; constexpr auto kFusionOpConv2DBackpropInputReluGradV2Name = "FusionOp_Conv2DBackpropInput_ReluGradV2";
constexpr auto kFusionOpConv2DBackpropInputAddNReluGradV2Name = "FusionOp_Conv2DBackpropInput_AddN_ReluGradV2"; constexpr auto kFusionOpConv2DBackpropInputAddNReluGradV2Name = "FusionOp_Conv2DBackpropInput_AddN_ReluGradV2";
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册