diff --git a/mindspore/lite/nnacl/infer/infer_register.c b/mindspore/lite/nnacl/infer/infer_register.c index d353b1f551..0f355c5e39 100644 --- a/mindspore/lite/nnacl/infer/infer_register.c +++ b/mindspore/lite/nnacl/infer/infer_register.c @@ -15,12 +15,364 @@ */ #include "nnacl/infer/infer_register.h" #ifdef MS_COMPILE_IOS +extern void _AudioSpectrogramPrimType_AudioSpectrogram(); +extern void _EqualPrimType_Equal(); +extern void _GreaterPrimType_Greater(); +extern void _GreaterEqualPrimType_GreaterEqual(); +extern void _LessPrimType_Less(); +extern void _LessEqualPrimType_LessEqual(); +extern void _NotEqualPrimType_NotEqual(); +extern void _DivGradPrimType_DivGrad(); +extern void _MulGradPrimType_MulGrad(); +extern void _MinimumGradPrimType_MinimumGrad(); +extern void _FullConnectionPrimType_FullConnection(); +extern void _AdamPrimType_Adam(); +extern void _SlicePrimType_SliceFusion(); +extern void _LSTMPrimType_LSTM(); +extern void _TilePrimType_TileFusion(); +extern void _EmbeddingLookupPrimType_EmbeddingLookupFusion(); +extern void _AssignAddPrimType_AssignAdd(); +extern void _LinSpacePrimType_LinSpace(); +extern void _AddGradPrimType_AddGrad(); +extern void _SubGradPrimType_SubGrad(); +extern void _OneHotPrimType_OneHot(); +extern void _RandomStandardNormalPrimType_RandomStandardNormal(); +extern void _StridedSlicePrimType_StridedSlice(); +extern void _FillPrimType_Fill(); +extern void _CastPrimType_Cast(); +extern void _SGDPrimType_SGD(); +extern void _UniquePrimType_Unique(); +extern void _BatchToSpacePrimType_BatchToSpace(); +extern void _TensorListFromTensorPrimType_TensorListFromTensor(); +extern void _TensorListGetItemPrimType_TensorListGetItem(); +extern void _SoftmaxCrossEntropyWithLogitsPrimType_SoftmaxCrossEntropyWithLogits(); +extern void _ExpandDimsPrimType_ExpandDims(); +extern void _InvertPermutationPrimType_InvertPermutation(); +extern void _MergePrimType_Merge(); +extern void _RfftPrimType_Rfft(); +extern void _Conv2DBackpropInputFusionPrimType_Conv2DBackpropInputFusion(); +extern void _ScatterNdPrimType_ScatterNd(); +extern void _FusedBatchNormPrimType_FusedBatchNorm(); +extern void _PartialPrimType_PartialFusion(); +extern void _HashtableLookupPrimType_HashtableLookup(); extern void _ReducePrimType_ReduceFusion(); +extern void _GatherPrimType_Gather(); +extern void _SplitPrimType_Split(); +extern void _RangePrimType_Range(); +extern void _BroadcastToPrimType_BroadcastTo(); +extern void _UnsortedSegmentSumPrimType_UnsortedSegmentSum(); +extern void _SqueezePrimType_Squeeze(); +extern void _MaximumGradPrimType_MaximumGrad(); +extern void _PowPrimType_PowFusion(); +extern void _PriorBoxPrimType_PriorBox(); +extern void _SpaceToDepthPrimType_SpaceToDepth(); +extern void _RankPrimType_Rank(); +extern void _MfccPrimType_Mfcc(); +extern void _BinaryCrossEntropyPrimType_BinaryCrossEntropy(); +extern void _Conv2DBackpropFilterFusionPrimType_Conv2DBackpropFilterFusion(); +extern void _WherePrimType_Where(); +extern void _DetectionPostProcessPrimType_DetectionPostProcess(); +extern void _FlattenGradPrimType_FlattenGrad(); +extern void _ConstantOfShapePrimType_ConstantOfShape(); +extern void _SelectPrimType_Select(); +extern void _ConcatPrimType_Concat(); +extern void _StackPrimType_Stack(); +extern void _SwitchPrimType_Switch(); +extern void _NonMaxSuppressionPrimType_NonMaxSuppression(); +extern void _SkipGramPrimType_SkipGram(); +extern void _AddNPrimType_AddN(); +extern void _ArgMinPrimType_ArgMinFusion(); +extern void _ArgMaxPrimType_ArgMaxFusion(); +extern void _LayerNormFusionPrimType_LayerNormFusion(); +extern void _MaxPoolPrimType_MaxPoolFusion(); +extern void _AvgPoolPrimType_AvgPoolFusion(); +extern void _AssignPrimType_Assign(); +extern void _CropPrimType_Crop(); +extern void _UnsqueezePrimType_Unsqueeze(); +extern void _AbsPrimType_Abs(); +extern void _AbsGradPrimType_AbsGrad(); +extern void _ActivationPrimType_Activation(); +extern void _ActivationGradPrimType_ActivationGrad(); +extern void _BatchNormPrimType_BatchNorm(); +extern void _BinaryCrossEntropyGradPrimType_BinaryCrossEntropyGrad(); +extern void _BiasAddPrimType_BiasAdd(); +extern void _CeilPrimType_Ceil(); +extern void _ClipPrimType_Clip(); +extern void _ControlDependPrimType_ControlDepend(); +extern void _CosPrimType_Cos(); +extern void _DependPrimType_Depend(); +extern void _EluPrimType_Elu(); +extern void _ErfPrimType_Erf(); +extern void _ExpPrimType_ExpFusion(); +extern void _FakeQuantWithMinMaxVarsPrimType_FakeQuantWithMinMaxVars(); +extern void _FloorPrimType_Floor(); +extern void _IfPrimType_If(); +extern void _InstanceNormPrimType_InstanceNorm(); +extern void _IsFinitePrimType_IsFinite(); +extern void _LeakyReluPrimType_LeakyRelu(); +extern void _LogPrimType_Log(); +extern void _LogGradPrimType_LogGrad(); +extern void _LogicalNotPrimType_LogicalNot(); +extern void _LRNPrimType_LRN(); +extern void _L2NormalizePrimType_L2NormalizeFusion(); +extern void _NegPrimType_Neg(); +extern void _NegGradPrimType_NegGrad(); +extern void _PowerGradPrimType_PowerGrad(); +extern void _PReLUPrimType_PReLUFusion(); +extern void _ReciprocalPrimType_Reciprocal(); +extern void _ReverseSequencePrimType_ReverseSequence(); +extern void _ReversePrimType_ReverseV2(); +extern void _RoundPrimType_Round(); +extern void _RsqrtPrimType_Rsqrt(); +extern void _ScalePrimType_ScaleFusion(); +extern void _SigmoidCrossEntropyWithLogitsPrimType_SigmoidCrossEntropyWithLogits(); +extern void _SigmoidCrossEntropyWithLogitsGradPrimType_SigmoidCrossEntropyWithLogitsGrad(); +extern void _SinPrimType_Sin(); +extern void _SmoothL1LossPrimType_SmoothL1Loss(); +extern void _SmoothL1LossGradPrimType_SmoothL1LossGrad(); +extern void _SqrtPrimType_Sqrt(); +extern void _SquarePrimType_Square(); +extern void _ZerosLikePrimType_ZerosLike(); +extern void _AssertPrimType_Assert(); +extern void _TensorListSetItemPrimType_TensorListSetItem(); +extern void _SplicePrimType_Splice(); +extern void _SparseSoftmaxCrossEntropyWithLogitsPrimType_SparseSoftmaxCrossEntropyWithLogits(); +extern void _GRUPrimType_GRU(); +extern void _SizeOpPrimType_Size(); +extern void _CustomExtractFeaturesPrimType_CustomExtractFeatures(); +extern void _CustomPredictPrimType_CustomPredict(); +extern void _WhilePrimType_While(); +extern void _StridedSliceGradPrimType_StridedSliceGrad(); +extern void _LshProjectionPrimType_LshProjection(); +extern void _SoftmaxPrimType_Softmax(); +extern void _CustomNormalizePrimType_CustomNormalize(); +extern void _UnstackPrimType_Unstack(); +extern void _ROIPoolingPrimType_ROIPooling(); +extern void _LayerNormGradPrimType_LayerNormGrad(); +extern void _DropoutGradPrimType_DropoutGrad(); +extern void _TopKPrimType_TopKFusion(); +extern void _ApplyMomentumPrimType_ApplyMomentum(); +extern void _AdderPrimType_AdderFusion(); +extern void _Conv2DPrimType_Conv2DFusion(); +extern void _UniformRealPrimType_UniformReal(); +extern void _AvgPoolGradPrimType_AvgPoolGrad(); +extern void _MaxPoolGradPrimType_MaxPoolGrad(); +extern void _DepthToSpacePrimType_DepthToSpace(); +extern void _Conv2dTransposePrimType_Conv2dTransposeFusion(); +extern void _QuantDTypeCastPrimType_QuantDTypeCast(); +extern void _FftImagPrimType_FftImag(); +extern void _FftRealPrimType_FftReal(); +extern void _ResizePrimType_Resize(); +extern void _SpaceToBatchNDPrimType_SpaceToBatchND(); +extern void _TransposePrimType_Transpose(); +extern void _TensorListReservePrimType_TensorListReserve(); +extern void _ShapePrimType_Shape(); +extern void _BiasAddGradPrimType_BiasAddGrad(); +extern void _MatMulPrimType_MatMul(); +extern void _DropoutPrimType_Dropout(); extern void _ReshapePrimType_Reshape(); +extern void _PadPrimType_PadFusion(); +extern void _TensorListStackPrimType_TensorListStack(); +extern void _FlattenPrimType_Flatten(); +extern void _BatchNormGradPrimType_BatchNormGrad(); +extern void _SpaceToBatchPrimType_SpaceToBatch(); +extern void _GatherNdPrimType_GatherNd(); +extern void _CropAndResizePrimType_CropAndResize(); +extern void _SparseToDensePrimType_SparseToDense(); +extern void _AddPrimType_AddFusion(); +extern void _DivPrimType_DivFusion(); +extern void _EltwisePrimType_Eltwise(); +extern void _FloorDivPrimType_FloorDiv(); +extern void _FloorModPrimType_FloorMod(); +extern void _LogicalAndPrimType_LogicalAnd(); +extern void _LogicalOrPrimType_LogicalOr(); +extern void _MaximumPrimType_Maximum(); +extern void _MinimumPrimType_Minimum(); +extern void _ModPrimType_Mod(); +extern void _MulPrimType_MulFusion(); +extern void _RealDivPrimType_RealDiv(); +extern void _SubPrimType_SubFusion(); +extern void _SquaredDifferencePrimType_SquaredDifference(); void RegisterInfer() { + _AudioSpectrogramPrimType_AudioSpectrogram(); + _EqualPrimType_Equal(); + _GreaterPrimType_Greater(); + _GreaterEqualPrimType_GreaterEqual(); + _LessPrimType_Less(); + _LessEqualPrimType_LessEqual(); + _NotEqualPrimType_NotEqual(); + _DivGradPrimType_DivGrad(); + _MulGradPrimType_MulGrad(); + _MinimumGradPrimType_MinimumGrad(); + _FullConnectionPrimType_FullConnection(); + _AdamPrimType_Adam(); + _SlicePrimType_SliceFusion(); + _LSTMPrimType_LSTM(); + _TilePrimType_TileFusion(); + _EmbeddingLookupPrimType_EmbeddingLookupFusion(); + _AssignAddPrimType_AssignAdd(); + _LinSpacePrimType_LinSpace(); + _AddGradPrimType_AddGrad(); + _SubGradPrimType_SubGrad(); + _OneHotPrimType_OneHot(); + _RandomStandardNormalPrimType_RandomStandardNormal(); + _StridedSlicePrimType_StridedSlice(); + _FillPrimType_Fill(); + _CastPrimType_Cast(); + _SGDPrimType_SGD(); + _UniquePrimType_Unique(); + _BatchToSpacePrimType_BatchToSpace(); + _TensorListFromTensorPrimType_TensorListFromTensor(); + _TensorListGetItemPrimType_TensorListGetItem(); + _SoftmaxCrossEntropyWithLogitsPrimType_SoftmaxCrossEntropyWithLogits(); + _ExpandDimsPrimType_ExpandDims(); + _InvertPermutationPrimType_InvertPermutation(); + _MergePrimType_Merge(); + _RfftPrimType_Rfft(); + _Conv2DBackpropInputFusionPrimType_Conv2DBackpropInputFusion(); + _ScatterNdPrimType_ScatterNd(); + _FusedBatchNormPrimType_FusedBatchNorm(); + _PartialPrimType_PartialFusion(); + _HashtableLookupPrimType_HashtableLookup(); _ReducePrimType_ReduceFusion(); + _GatherPrimType_Gather(); + _SplitPrimType_Split(); + _RangePrimType_Range(); + _BroadcastToPrimType_BroadcastTo(); + _UnsortedSegmentSumPrimType_UnsortedSegmentSum(); + _SqueezePrimType_Squeeze(); + _MaximumGradPrimType_MaximumGrad(); + _PowPrimType_PowFusion(); + _PriorBoxPrimType_PriorBox(); + _SpaceToDepthPrimType_SpaceToDepth(); + _RankPrimType_Rank(); + _MfccPrimType_Mfcc(); + _BinaryCrossEntropyPrimType_BinaryCrossEntropy(); + _Conv2DBackpropFilterFusionPrimType_Conv2DBackpropFilterFusion(); + _WherePrimType_Where(); + _DetectionPostProcessPrimType_DetectionPostProcess(); + _FlattenGradPrimType_FlattenGrad(); + _ConstantOfShapePrimType_ConstantOfShape(); + _SelectPrimType_Select(); + _ConcatPrimType_Concat(); + _StackPrimType_Stack(); + _SwitchPrimType_Switch(); + _NonMaxSuppressionPrimType_NonMaxSuppression(); + _SkipGramPrimType_SkipGram(); + _AddNPrimType_AddN(); + _ArgMinPrimType_ArgMinFusion(); + _ArgMaxPrimType_ArgMaxFusion(); + _LayerNormFusionPrimType_LayerNormFusion(); + _MaxPoolPrimType_MaxPoolFusion(); + _AvgPoolPrimType_AvgPoolFusion(); + _AssignPrimType_Assign(); + _CropPrimType_Crop(); + _UnsqueezePrimType_Unsqueeze(); + _AbsPrimType_Abs(); + _AbsGradPrimType_AbsGrad(); + _ActivationPrimType_Activation(); + _ActivationGradPrimType_ActivationGrad(); + _BatchNormPrimType_BatchNorm(); + _BinaryCrossEntropyGradPrimType_BinaryCrossEntropyGrad(); + _BiasAddPrimType_BiasAdd(); + _CeilPrimType_Ceil(); + _ClipPrimType_Clip(); + _ControlDependPrimType_ControlDepend(); + _CosPrimType_Cos(); + _DependPrimType_Depend(); + _EluPrimType_Elu(); + _ErfPrimType_Erf(); + _ExpPrimType_ExpFusion(); + _FakeQuantWithMinMaxVarsPrimType_FakeQuantWithMinMaxVars(); + _FloorPrimType_Floor(); + _IfPrimType_If(); + _InstanceNormPrimType_InstanceNorm(); + _IsFinitePrimType_IsFinite(); + _LeakyReluPrimType_LeakyRelu(); + _LogPrimType_Log(); + _LogGradPrimType_LogGrad(); + _LogicalNotPrimType_LogicalNot(); + _LRNPrimType_LRN(); + _L2NormalizePrimType_L2NormalizeFusion(); + _NegPrimType_Neg(); + _NegGradPrimType_NegGrad(); + _PowerGradPrimType_PowerGrad(); + _PReLUPrimType_PReLUFusion(); + _ReciprocalPrimType_Reciprocal(); + _ReverseSequencePrimType_ReverseSequence(); + _ReversePrimType_ReverseV2(); + _RoundPrimType_Round(); + _RsqrtPrimType_Rsqrt(); + _ScalePrimType_ScaleFusion(); + _SigmoidCrossEntropyWithLogitsPrimType_SigmoidCrossEntropyWithLogits(); + _SigmoidCrossEntropyWithLogitsGradPrimType_SigmoidCrossEntropyWithLogitsGrad(); + _SinPrimType_Sin(); + _SmoothL1LossPrimType_SmoothL1Loss(); + _SmoothL1LossGradPrimType_SmoothL1LossGrad(); + _SqrtPrimType_Sqrt(); + _SquarePrimType_Square(); + _ZerosLikePrimType_ZerosLike(); + _AssertPrimType_Assert(); + _TensorListSetItemPrimType_TensorListSetItem(); + _SplicePrimType_Splice(); + _SparseSoftmaxCrossEntropyWithLogitsPrimType_SparseSoftmaxCrossEntropyWithLogits(); + _GRUPrimType_GRU(); + _SizeOpPrimType_Size(); + _CustomExtractFeaturesPrimType_CustomExtractFeatures(); + _CustomPredictPrimType_CustomPredict(); + _WhilePrimType_While(); + _StridedSliceGradPrimType_StridedSliceGrad(); + _LshProjectionPrimType_LshProjection(); + _SoftmaxPrimType_Softmax(); + _CustomNormalizePrimType_CustomNormalize(); + _UnstackPrimType_Unstack(); + _ROIPoolingPrimType_ROIPooling(); + _LayerNormGradPrimType_LayerNormGrad(); + _DropoutGradPrimType_DropoutGrad(); + _TopKPrimType_TopKFusion(); + _ApplyMomentumPrimType_ApplyMomentum(); + _AdderPrimType_AdderFusion(); + _Conv2DPrimType_Conv2DFusion(); + _UniformRealPrimType_UniformReal(); + _AvgPoolGradPrimType_AvgPoolGrad(); + _MaxPoolGradPrimType_MaxPoolGrad(); + _DepthToSpacePrimType_DepthToSpace(); + _Conv2dTransposePrimType_Conv2dTransposeFusion(); + _QuantDTypeCastPrimType_QuantDTypeCast(); + _FftImagPrimType_FftImag(); + _FftRealPrimType_FftReal(); + _ResizePrimType_Resize(); + _SpaceToBatchNDPrimType_SpaceToBatchND(); + _TransposePrimType_Transpose(); + _TensorListReservePrimType_TensorListReserve(); + _ShapePrimType_Shape(); + _BiasAddGradPrimType_BiasAddGrad(); + _MatMulPrimType_MatMul(); + _DropoutPrimType_Dropout(); _ReshapePrimType_Reshape(); + _PadPrimType_PadFusion(); + _TensorListStackPrimType_TensorListStack(); + _FlattenPrimType_Flatten(); + _BatchNormGradPrimType_BatchNormGrad(); + _SpaceToBatchPrimType_SpaceToBatch(); + _GatherNdPrimType_GatherNd(); + _CropAndResizePrimType_CropAndResize(); + _SparseToDensePrimType_SparseToDense(); + _AddPrimType_AddFusion(); + _DivPrimType_DivFusion(); + _EltwisePrimType_Eltwise(); + _FloorDivPrimType_FloorDiv(); + _FloorModPrimType_FloorMod(); + _LogicalAndPrimType_LogicalAnd(); + _LogicalOrPrimType_LogicalOr(); + _MaximumPrimType_Maximum(); + _MinimumPrimType_Minimum(); + _ModPrimType_Mod(); + _MulPrimType_MulFusion(); + _RealDivPrimType_RealDiv(); + _SubPrimType_SubFusion(); + _SquaredDifferencePrimType_SquaredDifference(); } #endif InferShape *g_infer_func = NULL; diff --git a/mindspore/lite/src/ops/populate/activation_grad_populate.cc b/mindspore/lite/src/ops/populate/activation_grad_populate.cc index f832719648..cf25f440fd 100644 --- a/mindspore/lite/src/ops/populate/activation_grad_populate.cc +++ b/mindspore/lite/src/ops/populate/activation_grad_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32_grad/activation_grad.h" +using mindspore::schema::PrimitiveType_ActivationGrad; namespace mindspore { namespace lite { @@ -34,7 +35,6 @@ OpParameter *PopulateActivationGradParameter(const void *prim) { act_param->alpha_ = value->alpha(); return reinterpret_cast(act_param); } -Registry ActivationGradParameterRegistry(schema::PrimitiveType_ActivationGrad, PopulateActivationGradParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_ActivationGrad, PopulateActivationGradParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/activation_populate.cc b/mindspore/lite/src/ops/populate/activation_populate.cc index 1f0e8c9e01..3906f90b83 100644 --- a/mindspore/lite/src/ops/populate/activation_populate.cc +++ b/mindspore/lite/src/ops/populate/activation_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32/activation_fp32.h" +using mindspore::schema::PrimitiveType_Activation; namespace mindspore { namespace lite { @@ -37,6 +38,6 @@ OpParameter *PopulateRelu6Parameter(const void *prim) { } } // namespace -Registry g_relu6ParameterRegistry(schema::PrimitiveType_Activation, PopulateRelu6Parameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Activation, PopulateRelu6Parameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/adam_populate.cc b/mindspore/lite/src/ops/populate/adam_populate.cc index 38682785ee..61e6cc4151 100644 --- a/mindspore/lite/src/ops/populate/adam_populate.cc +++ b/mindspore/lite/src/ops/populate/adam_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_Adam; namespace mindspore { namespace lite { @@ -29,6 +30,6 @@ OpParameter *PopulateAdamParameter(const void *prim) { return param; } -Registry AdamParameterRegistry(schema::PrimitiveType_Adam, PopulateAdamParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Adam, PopulateAdamParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/add_populate.cc b/mindspore/lite/src/ops/populate/add_populate.cc index ff5ac84dc3..300a288ce8 100644 --- a/mindspore/lite/src/ops/populate/add_populate.cc +++ b/mindspore/lite/src/ops/populate/add_populate.cc @@ -16,6 +16,7 @@ #include "src/ops/populate/populate_register.h" #include "nnacl/arithmetic.h" #include "src/ops/populate/arithmetic_populate.h" +using mindspore::schema::PrimitiveType_AddFusion; namespace mindspore { namespace lite { @@ -33,6 +34,6 @@ OpParameter *PopulateAddParameter(const void *prim) { return reinterpret_cast(param); } } // namespace -Registry g_addParameterRegistry(schema::PrimitiveType_AddFusion, PopulateAddParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_AddFusion, PopulateAddParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/adder_populate.cc b/mindspore/lite/src/ops/populate/adder_populate.cc index 295a814306..b63977222c 100644 --- a/mindspore/lite/src/ops/populate/adder_populate.cc +++ b/mindspore/lite/src/ops/populate/adder_populate.cc @@ -16,6 +16,7 @@ #include "src/common/log_adapter.h" #include "nnacl/conv_parameter.h" #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_AdderFusion; namespace mindspore { namespace lite { @@ -57,6 +58,6 @@ OpParameter *PopulateAdderParameter(const void *prim) { } return reinterpret_cast(conv_param); } -Registry g_AdderParameterRegistry(schema::PrimitiveType_AdderFusion, PopulateAdderParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_AdderFusion, PopulateAdderParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/addn_populate.cc b/mindspore/lite/src/ops/populate/addn_populate.cc index 7932356f7a..4fc98b12b5 100644 --- a/mindspore/lite/src/ops/populate/addn_populate.cc +++ b/mindspore/lite/src/ops/populate/addn_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/op_base.h" +using mindspore::schema::PrimitiveType_AddN; namespace mindspore { namespace lite { @@ -31,6 +32,6 @@ OpParameter *PopulateAddNParameter(const void *prim) { return reinterpret_cast(addn_param); } } // namespace -Registry g_addNParameterRegistry(schema::PrimitiveType_AddN, PopulateAddNParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_AddN, PopulateAddNParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/argmax_populate.cc b/mindspore/lite/src/ops/populate/argmax_populate.cc index 99da4b97c1..3e322ade73 100644 --- a/mindspore/lite/src/ops/populate/argmax_populate.cc +++ b/mindspore/lite/src/ops/populate/argmax_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/arg_min_max_parameter.h" +using mindspore::schema::PrimitiveType_ArgMaxFusion; namespace mindspore { namespace lite { @@ -38,6 +39,6 @@ OpParameter *PopulateArgMaxParameter(const void *prim) { } } // namespace -Registry g_argMaxParameterRegistry(schema::PrimitiveType_ArgMaxFusion, PopulateArgMaxParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_ArgMaxFusion, PopulateArgMaxParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/argmin_populate.cc b/mindspore/lite/src/ops/populate/argmin_populate.cc index 630260526e..f56fd49e24 100644 --- a/mindspore/lite/src/ops/populate/argmin_populate.cc +++ b/mindspore/lite/src/ops/populate/argmin_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/arg_min_max_parameter.h" +using mindspore::schema::PrimitiveType_ArgMinFusion; namespace mindspore { namespace lite { @@ -38,6 +39,6 @@ OpParameter *PopulateArgMinParameter(const void *prim) { } } // namespace -Registry g_argMinParameterRegistry(schema::PrimitiveType_ArgMinFusion, PopulateArgMinParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_ArgMinFusion, PopulateArgMinParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/arithmetic_populate.cc b/mindspore/lite/src/ops/populate/arithmetic_populate.cc index f4bec62843..f646543b35 100644 --- a/mindspore/lite/src/ops/populate/arithmetic_populate.cc +++ b/mindspore/lite/src/ops/populate/arithmetic_populate.cc @@ -15,6 +15,23 @@ */ #include "src/ops/populate/arithmetic_populate.h" #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_Equal; +using mindspore::schema::PrimitiveType_FloorDiv; +using mindspore::schema::PrimitiveType_FloorMod; +using mindspore::schema::PrimitiveType_Greater; +using mindspore::schema::PrimitiveType_GreaterEqual; +using mindspore::schema::PrimitiveType_Less; +using mindspore::schema::PrimitiveType_LessEqual; +using mindspore::schema::PrimitiveType_LogicalAnd; +using mindspore::schema::PrimitiveType_LogicalOr; +using mindspore::schema::PrimitiveType_Maximum; +using mindspore::schema::PrimitiveType_MaximumGrad; +using mindspore::schema::PrimitiveType_Minimum; +using mindspore::schema::PrimitiveType_MinimumGrad; +using mindspore::schema::PrimitiveType_Mod; +using mindspore::schema::PrimitiveType_NotEqual; +using mindspore::schema::PrimitiveType_RealDiv; +using mindspore::schema::PrimitiveType_SquaredDifference; namespace mindspore { namespace lite { @@ -42,22 +59,22 @@ OpParameter *PopulateArithmetic(const void *primitive) { return reinterpret_cast(param); } -Registry g_MinimunGradParameterRegistry(schema::PrimitiveType_MinimumGrad, PopulateArithmetic, SCHEMA_CUR); -Registry g_MaximunGradParameterRegistry(schema::PrimitiveType_MaximumGrad, PopulateArithmetic, SCHEMA_CUR); -Registry g_realDivParameterRegistry(schema::PrimitiveType_RealDiv, PopulateArithmetic, SCHEMA_CUR); -Registry g_logicalAndParameterRegistry(schema::PrimitiveType_LogicalAnd, PopulateArithmetic, SCHEMA_CUR); -Registry g_parameterRegistry(schema::PrimitiveType_LogicalOr, PopulateArithmetic, SCHEMA_CUR); -Registry g_equalParameterRegistry(schema::PrimitiveType_Equal, PopulateArithmetic, SCHEMA_CUR); -Registry g_notEqualParameterRegistry(schema::PrimitiveType_NotEqual, PopulateArithmetic, SCHEMA_CUR); -Registry g_essParameterRegistry(schema::PrimitiveType_Less, PopulateArithmetic, SCHEMA_CUR); -Registry g_lessEqualParameterRegistry(schema::PrimitiveType_LessEqual, PopulateArithmetic, SCHEMA_CUR); -Registry g_greaterParameterRegistry(schema::PrimitiveType_Greater, PopulateArithmetic, SCHEMA_CUR); -Registry g_greaterEqualParameterRegistry(schema::PrimitiveType_GreaterEqual, PopulateArithmetic, SCHEMA_CUR); -Registry g_maximumParameterRegistry(schema::PrimitiveType_Maximum, PopulateArithmetic, SCHEMA_CUR); -Registry g_minimumParameterRegistry(schema::PrimitiveType_Minimum, PopulateArithmetic, SCHEMA_CUR); -Registry g_floorDivParameterRegistry(schema::PrimitiveType_FloorDiv, PopulateArithmetic, SCHEMA_CUR); -Registry g_floorModParameterRegistry(schema::PrimitiveType_FloorMod, PopulateArithmetic, SCHEMA_CUR); -Registry g_modParameterRegistry(schema::PrimitiveType_Mod, PopulateArithmetic, SCHEMA_CUR); -Registry g_squaredDifferenceParameterRegistry(schema::PrimitiveType_SquaredDifference, PopulateArithmetic, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_MinimumGrad, PopulateArithmetic, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_MaximumGrad, PopulateArithmetic, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_RealDiv, PopulateArithmetic, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_LogicalAnd, PopulateArithmetic, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_LogicalOr, PopulateArithmetic, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_Equal, PopulateArithmetic, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_NotEqual, PopulateArithmetic, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_Less, PopulateArithmetic, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_LessEqual, PopulateArithmetic, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_Greater, PopulateArithmetic, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_GreaterEqual, PopulateArithmetic, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_Maximum, PopulateArithmetic, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_Minimum, PopulateArithmetic, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_FloorDiv, PopulateArithmetic, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_FloorMod, PopulateArithmetic, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_Mod, PopulateArithmetic, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_SquaredDifference, PopulateArithmetic, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/arithmetic_self_populate.cc b/mindspore/lite/src/ops/populate/arithmetic_self_populate.cc index 99a47ebaaf..dad40f5b50 100644 --- a/mindspore/lite/src/ops/populate/arithmetic_self_populate.cc +++ b/mindspore/lite/src/ops/populate/arithmetic_self_populate.cc @@ -16,6 +16,21 @@ #include "src/common/log_adapter.h" #include "nnacl/arithmetic_self_parameter.h" #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_Abs; +using mindspore::schema::PrimitiveType_Ceil; +using mindspore::schema::PrimitiveType_Cos; +using mindspore::schema::PrimitiveType_Floor; +using mindspore::schema::PrimitiveType_Log; +using mindspore::schema::PrimitiveType_LogGrad; +using mindspore::schema::PrimitiveType_LogicalNot; +using mindspore::schema::PrimitiveType_Neg; +using mindspore::schema::PrimitiveType_NegGrad; +using mindspore::schema::PrimitiveType_Reciprocal; +using mindspore::schema::PrimitiveType_Round; +using mindspore::schema::PrimitiveType_Rsqrt; +using mindspore::schema::PrimitiveType_Sin; +using mindspore::schema::PrimitiveType_Sqrt; +using mindspore::schema::PrimitiveType_Square; namespace mindspore { namespace lite { @@ -32,20 +47,20 @@ OpParameter *PopulateArithmeticSelf(const void *prim) { return reinterpret_cast(arithmetic_self_param); } -Registry g_absParameterRegistry(schema::PrimitiveType_Abs, PopulateArithmeticSelf, SCHEMA_CUR); -Registry g_cosParameterRegistry(schema::PrimitiveType_Cos, PopulateArithmeticSelf, SCHEMA_CUR); -Registry g_sinParameterRegistry(schema::PrimitiveType_Sin, PopulateArithmeticSelf, SCHEMA_CUR); -Registry g_logParameterRegistry(schema::PrimitiveType_Log, PopulateArithmeticSelf, SCHEMA_CUR); -Registry g_negParameterRegistry(schema::PrimitiveType_Neg, PopulateArithmeticSelf, SCHEMA_CUR); -Registry g_negGradParameterRegistry(schema::PrimitiveType_NegGrad, PopulateArithmeticSelf, SCHEMA_CUR); -Registry g_logGradParameterRegistry(schema::PrimitiveType_LogGrad, PopulateArithmeticSelf, SCHEMA_CUR); -Registry g_sqrtParameterRegistry(schema::PrimitiveType_Sqrt, PopulateArithmeticSelf, SCHEMA_CUR); -Registry g_squareParameterRegistry(schema::PrimitiveType_Square, PopulateArithmeticSelf, SCHEMA_CUR); -Registry g_rsqrtParameterRegistry(schema::PrimitiveType_Rsqrt, PopulateArithmeticSelf, SCHEMA_CUR); -Registry g_logicalNotParameterRegistry(schema::PrimitiveType_LogicalNot, PopulateArithmeticSelf, SCHEMA_CUR); -Registry g_floorParameterRegistry(schema::PrimitiveType_Floor, PopulateArithmeticSelf, SCHEMA_CUR); -Registry g_ceilParameterRegistry(schema::PrimitiveType_Ceil, PopulateArithmeticSelf, SCHEMA_CUR); -Registry g_roundParameterRegistry(schema::PrimitiveType_Round, PopulateArithmeticSelf, SCHEMA_CUR); -Registry g_reciprocalParameterRegistry(schema::PrimitiveType_Reciprocal, PopulateArithmeticSelf, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Abs, PopulateArithmeticSelf, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_Cos, PopulateArithmeticSelf, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_Sin, PopulateArithmeticSelf, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_Log, PopulateArithmeticSelf, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_Neg, PopulateArithmeticSelf, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_NegGrad, PopulateArithmeticSelf, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_LogGrad, PopulateArithmeticSelf, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_Sqrt, PopulateArithmeticSelf, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_Square, PopulateArithmeticSelf, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_Rsqrt, PopulateArithmeticSelf, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_LogicalNot, PopulateArithmeticSelf, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_Floor, PopulateArithmeticSelf, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_Ceil, PopulateArithmeticSelf, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_Round, PopulateArithmeticSelf, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_Reciprocal, PopulateArithmeticSelf, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/assert_populate.cc b/mindspore/lite/src/ops/populate/assert_populate.cc index 3a83b0f714..2fa8c5e6df 100644 --- a/mindspore/lite/src/ops/populate/assert_populate.cc +++ b/mindspore/lite/src/ops/populate/assert_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_Assert; namespace mindspore { namespace lite { @@ -30,6 +31,6 @@ OpParameter *PopulateAssertParameter(const void *prim) { return reinterpret_cast(assert_parameter); } -Registry AssertParameterRegistry(schema::PrimitiveType_Assert, PopulateAssertParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Assert, PopulateAssertParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/assign_add_populate.cc b/mindspore/lite/src/ops/populate/assign_add_populate.cc index 3e601f6fb6..cd30b00c5d 100644 --- a/mindspore/lite/src/ops/populate/assign_add_populate.cc +++ b/mindspore/lite/src/ops/populate/assign_add_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_AssignAdd; namespace mindspore { namespace lite { @@ -29,6 +30,6 @@ OpParameter *PopulateAssignAddParameter(const void *prim) { return param; } -Registry AssignAddParameterRegistry(schema::PrimitiveType_AssignAdd, PopulateAssignAddParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_AssignAdd, PopulateAssignAddParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/assign_populate.cc b/mindspore/lite/src/ops/populate/assign_populate.cc index 191897af93..c20951541a 100644 --- a/mindspore/lite/src/ops/populate/assign_populate.cc +++ b/mindspore/lite/src/ops/populate/assign_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_Assign; namespace mindspore { namespace lite { @@ -30,6 +31,6 @@ OpParameter *PopulateAssignParameter(const void *prim) { return param; } -Registry AssignParameterRegistry(schema::PrimitiveType_Assign, PopulateAssignParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Assign, PopulateAssignParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/audio_spectrogram_populate.cc b/mindspore/lite/src/ops/populate/audio_spectrogram_populate.cc index 3c88e961c2..64e1d2dfb6 100644 --- a/mindspore/lite/src/ops/populate/audio_spectrogram_populate.cc +++ b/mindspore/lite/src/ops/populate/audio_spectrogram_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/infer/audio_spectrogram_infer.h" +using mindspore::schema::PrimitiveType_AudioSpectrogram; namespace mindspore { namespace lite { @@ -36,7 +37,6 @@ OpParameter *PopulateAudioSpectrogramParameter(const void *prim) { } } // namespace -Registry g_audioSpectrogramParameterRegistry(schema::PrimitiveType_AudioSpectrogram, PopulateAudioSpectrogramParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_AudioSpectrogram, PopulateAudioSpectrogramParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/batch_norm_populate.cc b/mindspore/lite/src/ops/populate/batch_norm_populate.cc index 77eb009a2e..d5a8abf5d3 100644 --- a/mindspore/lite/src/ops/populate/batch_norm_populate.cc +++ b/mindspore/lite/src/ops/populate/batch_norm_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/batchnorm_parameter.h" +using mindspore::schema::PrimitiveType_BatchNorm; namespace mindspore { namespace lite { @@ -35,7 +36,6 @@ OpParameter *PopulateBatchNorm(const void *prim) { } } // namespace -Registry g_batchNormParameterRegistry(schema::PrimitiveType_BatchNorm, PopulateBatchNorm, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_BatchNorm, PopulateBatchNorm, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/batch_to_space_populate.cc b/mindspore/lite/src/ops/populate/batch_to_space_populate.cc index f39744918f..387ed781e9 100644 --- a/mindspore/lite/src/ops/populate/batch_to_space_populate.cc +++ b/mindspore/lite/src/ops/populate/batch_to_space_populate.cc @@ -15,6 +15,8 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/batch_to_space.h" +using mindspore::schema::PrimitiveType_BatchToSpace; +using mindspore::schema::PrimitiveType_BatchToSpaceND; namespace mindspore { namespace lite { @@ -63,7 +65,7 @@ OpParameter *PopulateBatchToSpaceParameter(const void *prim) { return reinterpret_cast(batch_space_param); } } // namespace -Registry g_batchToSpaceRegistry(schema::PrimitiveType_BatchToSpace, PopulateBatchToSpaceParameter, SCHEMA_CUR); -Registry g_batchToSpaceNDRegistry(schema::PrimitiveType_BatchToSpaceND, PopulateBatchToSpaceParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_BatchToSpace, PopulateBatchToSpaceParameter, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_BatchToSpaceND, PopulateBatchToSpaceParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/bias_add_populate.cc b/mindspore/lite/src/ops/populate/bias_add_populate.cc index b58b222845..97f4bd005b 100644 --- a/mindspore/lite/src/ops/populate/bias_add_populate.cc +++ b/mindspore/lite/src/ops/populate/bias_add_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/arithmetic.h" +using mindspore::schema::PrimitiveType_BiasAdd; namespace mindspore { namespace lite { @@ -32,7 +33,6 @@ OpParameter *PopulateBiasAddParameter(const void *prim) { return reinterpret_cast(arithmetic_param); } } // namespace -Registry g_biasAddParameterRegistry(schema::PrimitiveType_BiasAdd, PopulateBiasAddParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_BiasAdd, PopulateBiasAddParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/bias_grad_populate.cc b/mindspore/lite/src/ops/populate/bias_grad_populate.cc index 72cb7a9d82..1caae879e8 100644 --- a/mindspore/lite/src/ops/populate/bias_grad_populate.cc +++ b/mindspore/lite/src/ops/populate/bias_grad_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/arithmetic.h" +using mindspore::schema::PrimitiveType_BiasAddGrad; namespace mindspore { namespace lite { @@ -33,7 +34,6 @@ OpParameter *PopulateBiasAddGradParameter(const void *prim) { } } // namespace -Registry g_populateBiasGradParameterParameterRegistry(schema::PrimitiveType_BiasAddGrad, PopulateBiasAddGradParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_BiasAddGrad, PopulateBiasAddGradParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/binary_cross_entropy_grad_populate.cc b/mindspore/lite/src/ops/populate/binary_cross_entropy_grad_populate.cc index 910b055460..b681bf05ca 100644 --- a/mindspore/lite/src/ops/populate/binary_cross_entropy_grad_populate.cc +++ b/mindspore/lite/src/ops/populate/binary_cross_entropy_grad_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32_grad/binary_cross_entropy_grad.h" +using mindspore::schema::PrimitiveType_BinaryCrossEntropyGrad; namespace mindspore { namespace lite { @@ -35,7 +36,6 @@ OpParameter *PopulateBinaryCrossEntropyGradParameter(const void *prim) { } } // namespace -Registry g_binaryCrossEntropyGradParameterRegistry(schema::PrimitiveType_BinaryCrossEntropyGrad, - PopulateBinaryCrossEntropyGradParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_BinaryCrossEntropyGrad, PopulateBinaryCrossEntropyGradParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/binary_cross_entropy_populate.cc b/mindspore/lite/src/ops/populate/binary_cross_entropy_populate.cc index 2ade2b29fc..1c90d4c985 100644 --- a/mindspore/lite/src/ops/populate/binary_cross_entropy_populate.cc +++ b/mindspore/lite/src/ops/populate/binary_cross_entropy_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32_grad/binary_cross_entropy.h" +using mindspore::schema::PrimitiveType_BinaryCrossEntropy; namespace mindspore { namespace lite { @@ -33,7 +34,6 @@ OpParameter *PopulateBinaryCrossEntropyParameter(const void *prim) { return reinterpret_cast(bce_param); } -Registry BinaryCrossEntropyParameterRegistry(schema::PrimitiveType_BinaryCrossEntropy, - PopulateBinaryCrossEntropyParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_BinaryCrossEntropy, PopulateBinaryCrossEntropyParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/broadcast_to_populate.cc b/mindspore/lite/src/ops/populate/broadcast_to_populate.cc index 0bc8dc4203..33d8817d5b 100644 --- a/mindspore/lite/src/ops/populate/broadcast_to_populate.cc +++ b/mindspore/lite/src/ops/populate/broadcast_to_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32/broadcast_to_fp32.h" +using mindspore::schema::PrimitiveType_BroadcastTo; namespace mindspore { namespace lite { @@ -37,6 +38,6 @@ OpParameter *PopulateBroadcastToParameter(const void *prim) { return reinterpret_cast(broadcast_param); } -Registry BroadcastToParameterRegistry(schema::PrimitiveType_BroadcastTo, PopulateBroadcastToParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_BroadcastTo, PopulateBroadcastToParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/cast_populate.cc b/mindspore/lite/src/ops/populate/cast_populate.cc index d30859aa66..89059e4729 100644 --- a/mindspore/lite/src/ops/populate/cast_populate.cc +++ b/mindspore/lite/src/ops/populate/cast_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_Cast; namespace mindspore { namespace lite { @@ -31,6 +32,6 @@ OpParameter *PopulateCastParameter(const void *prim) { } } // namespace -Registry g_castParameterRegistry(schema::PrimitiveType_Cast, PopulateCastParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Cast, PopulateCastParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/clip_populate.cc b/mindspore/lite/src/ops/populate/clip_populate.cc index 6a2f2e2078..c2b7694af9 100644 --- a/mindspore/lite/src/ops/populate/clip_populate.cc +++ b/mindspore/lite/src/ops/populate/clip_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_Clip; namespace mindspore { namespace lite { @@ -31,7 +32,6 @@ OpParameter *PopulateClipParameter(const void *prim) { } } // namespace -Registry g_clipParameterRegistry(schema::PrimitiveType_Clip, PopulateClipParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_Clip, PopulateClipParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/common_populate.cc b/mindspore/lite/src/ops/populate/common_populate.cc index dce88c778e..55bb3871f2 100644 --- a/mindspore/lite/src/ops/populate/common_populate.cc +++ b/mindspore/lite/src/ops/populate/common_populate.cc @@ -14,6 +14,8 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_Depend; +using mindspore::schema::PrimitiveType_ZerosLike; namespace mindspore { namespace lite { @@ -31,7 +33,7 @@ OpParameter *PopulateCommonParameter(const void *prim) { } } // namespace -Registry g_zerosLikeParameterRegistry(schema::PrimitiveType_ZerosLike, PopulateCommonParameter, SCHEMA_CUR); -Registry g_dependParameterRegistry(schema::PrimitiveType_Depend, PopulateCommonParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_ZerosLike, PopulateCommonParameter, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_Depend, PopulateCommonParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/concat_populate.cc b/mindspore/lite/src/ops/populate/concat_populate.cc index 10d0451675..c60e9dd0ae 100644 --- a/mindspore/lite/src/ops/populate/concat_populate.cc +++ b/mindspore/lite/src/ops/populate/concat_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/concat_parameter.h" +using mindspore::schema::PrimitiveType_Concat; namespace mindspore { namespace lite { @@ -33,6 +34,6 @@ OpParameter *PopulateConcatParameter(const void *prim) { } } // namespace -Registry g_concatParameterRegistry(schema::PrimitiveType_Concat, PopulateConcatParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Concat, PopulateConcatParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/constant_of_shape_populate.cc b/mindspore/lite/src/ops/populate/constant_of_shape_populate.cc index a6c89a795d..21acc21e70 100644 --- a/mindspore/lite/src/ops/populate/constant_of_shape_populate.cc +++ b/mindspore/lite/src/ops/populate/constant_of_shape_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/constant_of_shape_parameter.h" +using mindspore::schema::PrimitiveType_ConstantOfShape; namespace mindspore::lite { namespace { @@ -46,8 +47,7 @@ OpParameter *PopulateConstantOfShapeParameter(const void *prim) { } } return reinterpret_cast(param); +} } // namespace -Registry g_constantOfShapeParameterRegistry(schema::PrimitiveType_ConstantOfShape, PopulateConstantOfShapeParameter, - SCHEMA_CUR); -} // namespace +REG_POPULATE(PrimitiveType_ConstantOfShape, PopulateConstantOfShapeParameter, SCHEMA_CUR); } // namespace mindspore::lite diff --git a/mindspore/lite/src/ops/populate/conv2d_populate.cc b/mindspore/lite/src/ops/populate/conv2d_populate.cc index 4500cabbfb..d487efee69 100644 --- a/mindspore/lite/src/ops/populate/conv2d_populate.cc +++ b/mindspore/lite/src/ops/populate/conv2d_populate.cc @@ -16,6 +16,7 @@ #include "nnacl/conv_parameter.h" #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_Conv2DFusion; namespace mindspore { namespace lite { @@ -74,6 +75,6 @@ OpParameter *PopulateConvParameter(const void *prim) { return reinterpret_cast(conv_param); } } // namespace -Registry g_conv2DParameterRegistry(schema::PrimitiveType_Conv2DFusion, PopulateConvParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Conv2DFusion, PopulateConvParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/crop_and_resize_populate.cc b/mindspore/lite/src/ops/populate/crop_and_resize_populate.cc index 0908af070b..73534795a6 100644 --- a/mindspore/lite/src/ops/populate/crop_and_resize_populate.cc +++ b/mindspore/lite/src/ops/populate/crop_and_resize_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/resize_parameter.h" +using mindspore::schema::PrimitiveType_CropAndResize; namespace mindspore { namespace lite { namespace { @@ -35,7 +36,6 @@ OpParameter *PopulateCropAndResizeParameter(const void *prim) { } } // namespace -Registry g_cropAndResizeParameterRegistry(schema::PrimitiveType_CropAndResize, PopulateCropAndResizeParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_CropAndResize, PopulateCropAndResizeParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/crop_populate.cc b/mindspore/lite/src/ops/populate/crop_populate.cc index bdfb0ed193..1d5660270b 100644 --- a/mindspore/lite/src/ops/populate/crop_populate.cc +++ b/mindspore/lite/src/ops/populate/crop_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/crop_parameter.h" +using mindspore::schema::PrimitiveType_Crop; namespace mindspore { namespace lite { @@ -43,6 +44,6 @@ OpParameter *PopulateCropParameter(const void *prim) { } } // namespace -Registry g_cropParameterRegistry(schema::PrimitiveType_Crop, PopulateCropParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Crop, PopulateCropParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/custom_extract_features_populate.cc b/mindspore/lite/src/ops/populate/custom_extract_features_populate.cc index d7ab450664..73b652c2c3 100644 --- a/mindspore/lite/src/ops/populate/custom_extract_features_populate.cc +++ b/mindspore/lite/src/ops/populate/custom_extract_features_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_CustomExtractFeatures; namespace mindspore { namespace lite { @@ -30,7 +31,6 @@ OpParameter *PopulateExtractFeaturesParameter(const void *prim) { return param; } } // namespace -Registry g_customExtractFeaturesParameterRegistry(schema::PrimitiveType_CustomExtractFeatures, - PopulateExtractFeaturesParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_CustomExtractFeatures, PopulateExtractFeaturesParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/custom_normalize_populate.cc b/mindspore/lite/src/ops/populate/custom_normalize_populate.cc index 94fa7975ea..93427a2f38 100644 --- a/mindspore/lite/src/ops/populate/custom_normalize_populate.cc +++ b/mindspore/lite/src/ops/populate/custom_normalize_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_CustomNormalize; namespace mindspore { namespace lite { @@ -29,8 +30,7 @@ OpParameter *PopulateCustomNormalizeParameter(const void *prim) { param->type_ = primitive->value_type(); return param; } -Registry CustomNormalizeParameterRegistry(schema::PrimitiveType_CustomNormalize, PopulateCustomNormalizeParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_CustomNormalize, PopulateCustomNormalizeParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/custom_predict_populate.cc b/mindspore/lite/src/ops/populate/custom_predict_populate.cc index d0084f246c..6fe70800bf 100644 --- a/mindspore/lite/src/ops/populate/custom_predict_populate.cc +++ b/mindspore/lite/src/ops/populate/custom_predict_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/predict_parameter.h" +using mindspore::schema::PrimitiveType_CustomPredict; namespace mindspore { namespace lite { @@ -33,8 +34,7 @@ OpParameter *PopulateCustomPredictParameter(const void *prim) { param->weight_threshold = value->weight_threshold(); return reinterpret_cast(param); } -Registry CustomPredictParameterRegistry(schema::PrimitiveType_CustomPredict, PopulateCustomPredictParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_CustomPredict, PopulateCustomPredictParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/deconv2d_populate.cc b/mindspore/lite/src/ops/populate/deconv2d_populate.cc index f67111be23..b6d6237c1a 100644 --- a/mindspore/lite/src/ops/populate/deconv2d_populate.cc +++ b/mindspore/lite/src/ops/populate/deconv2d_populate.cc @@ -16,6 +16,7 @@ #include "src/common/log_adapter.h" #include "src/ops/populate/populate_register.h" #include "nnacl/conv_parameter.h" +using mindspore::schema::PrimitiveType_Conv2dTransposeFusion; namespace mindspore { namespace lite { @@ -77,6 +78,6 @@ OpParameter *PopulateDeconvParameter(const void *prim) { return reinterpret_cast(conv_param); } -Registry g_Deconv2DParameterRegistry(schema::PrimitiveType_Conv2dTransposeFusion, PopulateDeconvParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Conv2dTransposeFusion, PopulateDeconvParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/depth_to_space_populate.cc b/mindspore/lite/src/ops/populate/depth_to_space_populate.cc index fe6b92a785..27df69abab 100644 --- a/mindspore/lite/src/ops/populate/depth_to_space_populate.cc +++ b/mindspore/lite/src/ops/populate/depth_to_space_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/depth_to_space_parameter.h" +using mindspore::schema::PrimitiveType_DepthToSpace; namespace mindspore { namespace lite { @@ -35,6 +36,6 @@ OpParameter *PopulateDepthToSpaceParameter(const void *prim) { } } // namespace -Registry g_depthToSpaceParamRegistry(schema::PrimitiveType_DepthToSpace, PopulateDepthToSpaceParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_DepthToSpace, PopulateDepthToSpaceParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/detection_post_process_populate.cc b/mindspore/lite/src/ops/populate/detection_post_process_populate.cc index 4e66d25374..db22340839 100644 --- a/mindspore/lite/src/ops/populate/detection_post_process_populate.cc +++ b/mindspore/lite/src/ops/populate/detection_post_process_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/detection_post_process_parameter.h" +using mindspore::schema::PrimitiveType_DetectionPostProcess; namespace mindspore { namespace lite { @@ -44,8 +45,7 @@ OpParameter *PopulateDetectionPostProcessParameter(const void *prim) { return reinterpret_cast(detection_post_process_parameter); } } // namespace -Registry g_detectionPostProcessParameterRegistry(schema::PrimitiveType_DetectionPostProcess, - PopulateDetectionPostProcessParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_DetectionPostProcess, PopulateDetectionPostProcessParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/div_populate.cc b/mindspore/lite/src/ops/populate/div_populate.cc index 647ef86947..b686fc1c4a 100644 --- a/mindspore/lite/src/ops/populate/div_populate.cc +++ b/mindspore/lite/src/ops/populate/div_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "src/ops/populate/arithmetic_populate.h" +using mindspore::schema::PrimitiveType_DivFusion; namespace mindspore { namespace lite { @@ -27,7 +28,6 @@ OpParameter *PopulateDivParameter(const void *prim) { return reinterpret_cast(param); } -Registry g_divParameterRegistry(schema::PrimitiveType_DivFusion, PopulateDivParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_DivFusion, PopulateDivParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/eltwise_populate.cc b/mindspore/lite/src/ops/populate/eltwise_populate.cc index b58ed8f865..5b4b2d731f 100644 --- a/mindspore/lite/src/ops/populate/eltwise_populate.cc +++ b/mindspore/lite/src/ops/populate/eltwise_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "src/ops/populate/arithmetic_populate.h" +using mindspore::schema::PrimitiveType_Eltwise; namespace mindspore { namespace lite { @@ -31,6 +32,6 @@ OpParameter *PopulateEltwiseParameter(const void *prim) { } } // namespace -Registry g_eltwiseParameterRegistry(schema::PrimitiveType_Eltwise, PopulateEltwiseParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Eltwise, PopulateEltwiseParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/elu_populate.cc b/mindspore/lite/src/ops/populate/elu_populate.cc index b5854f3b6e..c99c275e03 100644 --- a/mindspore/lite/src/ops/populate/elu_populate.cc +++ b/mindspore/lite/src/ops/populate/elu_populate.cc @@ -15,6 +15,7 @@ */ #include "nnacl/fp32/elu_fp32.h" #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_Elu; namespace mindspore { namespace lite { @@ -33,6 +34,6 @@ OpParameter *PopulateEluParameter(const void *prim) { return reinterpret_cast(elu_parameter); } } // namespace -Registry g_eluParameterRegistry(schema::PrimitiveType_Elu, PopulateEluParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Elu, PopulateEluParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/embedding_lookup_populate.cc b/mindspore/lite/src/ops/populate/embedding_lookup_populate.cc index 3c0881d1bc..95e35e47dd 100644 --- a/mindspore/lite/src/ops/populate/embedding_lookup_populate.cc +++ b/mindspore/lite/src/ops/populate/embedding_lookup_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32/embedding_lookup_fp32.h" +using mindspore::schema::PrimitiveType_EmbeddingLookupFusion; namespace mindspore { namespace lite { @@ -40,8 +41,7 @@ OpParameter *PopulateEmbeddingLookupParameter(const void *prim) { return reinterpret_cast(param); } -Registry EmbeddingLookupParameterRegistry(schema::PrimitiveType_EmbeddingLookupFusion, PopulateEmbeddingLookupParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_EmbeddingLookupFusion, PopulateEmbeddingLookupParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/erf_populate.cc b/mindspore/lite/src/ops/populate/erf_populate.cc index ea7657b118..51593146b0 100644 --- a/mindspore/lite/src/ops/populate/erf_populate.cc +++ b/mindspore/lite/src/ops/populate/erf_populate.cc @@ -15,9 +15,10 @@ */ #include "src/ops/populate/populate_register.h" #include "src/ops/populate/default_populate.h" +using mindspore::schema::PrimitiveType_Erf; namespace mindspore { namespace lite { -Registry g_erfParameterRegistry(schema::PrimitiveType_Erf, DefaultPopulateParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Erf, DefaultPopulateParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/exp_populate.cc b/mindspore/lite/src/ops/populate/exp_populate.cc index 122dc3c84a..f7a3e486c0 100644 --- a/mindspore/lite/src/ops/populate/exp_populate.cc +++ b/mindspore/lite/src/ops/populate/exp_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32/exp_fp32.h" +using mindspore::schema::PrimitiveType_ExpFusion; namespace mindspore { namespace lite { @@ -40,6 +41,6 @@ OpParameter *PopulateExpParameter(const void *prim) { return reinterpret_cast(exp_parameter); } -Registry ExpParameterRegistry(schema::PrimitiveType_ExpFusion, PopulateExpParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_ExpFusion, PopulateExpParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/expand_dims_populate.cc b/mindspore/lite/src/ops/populate/expand_dims_populate.cc index 47351d005f..b63bf7cf80 100644 --- a/mindspore/lite/src/ops/populate/expand_dims_populate.cc +++ b/mindspore/lite/src/ops/populate/expand_dims_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_ExpandDims; namespace mindspore { namespace lite { @@ -31,6 +32,6 @@ OpParameter *PopulateExpandDimsParameter(const void *prim) { } } // namespace -Registry g_expandDimsParameterRegistry(schema::PrimitiveType_ExpandDims, PopulateExpandDimsParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_ExpandDims, PopulateExpandDimsParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/fill_populate.cc b/mindspore/lite/src/ops/populate/fill_populate.cc index b48f4b24cf..725b17f453 100644 --- a/mindspore/lite/src/ops/populate/fill_populate.cc +++ b/mindspore/lite/src/ops/populate/fill_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_Fill; namespace mindspore { namespace lite { @@ -31,7 +32,6 @@ OpParameter *PopulateFillParameter(const void *prim) { } } // namespace -Registry g_fillParameterRegistry(schema::PrimitiveType_Fill, PopulateFillParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_Fill, PopulateFillParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/flatten_populate.cc b/mindspore/lite/src/ops/populate/flatten_populate.cc index 09cbab37cd..1ff2994b80 100644 --- a/mindspore/lite/src/ops/populate/flatten_populate.cc +++ b/mindspore/lite/src/ops/populate/flatten_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_Flatten; namespace mindspore { namespace lite { @@ -30,7 +31,6 @@ OpParameter *PopulateFlattenParameter(const void *prim) { return reinterpret_cast(flatten_param); } -Registry FlattenParameterRegistry(schema::PrimitiveType_Flatten, PopulateFlattenParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_Flatten, PopulateFlattenParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/full_connection_populate.cc b/mindspore/lite/src/ops/populate/full_connection_populate.cc index e60c48a222..b8fdc0afef 100644 --- a/mindspore/lite/src/ops/populate/full_connection_populate.cc +++ b/mindspore/lite/src/ops/populate/full_connection_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/matmul_parameter.h" +using mindspore::schema::PrimitiveType_FullConnection; namespace mindspore { namespace lite { @@ -45,6 +46,6 @@ OpParameter *PopulateFullconnectionParameter(const void *prim) { } } // namespace -Registry g_fullConnRegistry(schema::PrimitiveType_FullConnection, PopulateFullconnectionParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_FullConnection, PopulateFullconnectionParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/fused_batchnorm_populate.cc b/mindspore/lite/src/ops/populate/fused_batchnorm_populate.cc index 1457eeea6a..e80ae840cf 100644 --- a/mindspore/lite/src/ops/populate/fused_batchnorm_populate.cc +++ b/mindspore/lite/src/ops/populate/fused_batchnorm_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/batchnorm_parameter.h" +using mindspore::schema::PrimitiveType_FusedBatchNorm; namespace mindspore { namespace lite { @@ -35,7 +36,6 @@ OpParameter *PopulateFusedBatchNorm(const void *prim) { return reinterpret_cast(batch_norm_param); } -Registry FusedBatchNormParameterRegistry(schema::PrimitiveType_FusedBatchNorm, PopulateFusedBatchNorm, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_FusedBatchNorm, PopulateFusedBatchNorm, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/gather_nd_populate.cc b/mindspore/lite/src/ops/populate/gather_nd_populate.cc index 953cfe1b37..1d78ae613b 100644 --- a/mindspore/lite/src/ops/populate/gather_nd_populate.cc +++ b/mindspore/lite/src/ops/populate/gather_nd_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32/gatherNd_fp32.h" +using mindspore::schema::PrimitiveType_GatherNd; namespace mindspore { namespace lite { @@ -32,6 +33,6 @@ OpParameter *PopulateGatherNdParameter(const void *prim) { } } // namespace -Registry g_gatherNdParameterRegistry(schema::PrimitiveType_GatherNd, PopulateGatherNdParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_GatherNd, PopulateGatherNdParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/gather_populate.cc b/mindspore/lite/src/ops/populate/gather_populate.cc index 0da4594501..73c2ad7568 100644 --- a/mindspore/lite/src/ops/populate/gather_populate.cc +++ b/mindspore/lite/src/ops/populate/gather_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/gather_parameter.h" +using mindspore::schema::PrimitiveType_Gather; namespace mindspore { namespace lite { @@ -33,6 +34,6 @@ OpParameter *PopulateGatherParameter(const void *prim) { } } // namespace -Registry g_gatherParameterRegistry(schema::PrimitiveType_Gather, PopulateGatherParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Gather, PopulateGatherParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/gru_populate.cc b/mindspore/lite/src/ops/populate/gru_populate.cc index 4e492aca98..f05e8f31d0 100644 --- a/mindspore/lite/src/ops/populate/gru_populate.cc +++ b/mindspore/lite/src/ops/populate/gru_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32/gru_fp32.h" +using mindspore::schema::PrimitiveType_GRU; namespace mindspore { namespace lite { @@ -39,6 +40,6 @@ OpParameter *PopulateGruParameter(const void *prim) { } } // namespace -Registry g_gruParameterRegistry(schema::PrimitiveType_GRU, PopulateGruParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_GRU, PopulateGruParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/hashtable_lookup_populate.cc b/mindspore/lite/src/ops/populate/hashtable_lookup_populate.cc index d4ff5ac11e..164b9611f1 100644 --- a/mindspore/lite/src/ops/populate/hashtable_lookup_populate.cc +++ b/mindspore/lite/src/ops/populate/hashtable_lookup_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_HashtableLookup; namespace mindspore { namespace lite { @@ -29,8 +30,7 @@ OpParameter *PopulateHashtableLookupParameter(const void *prim) { param->type_ = primitive->value_type(); return param; } -Registry HashtableLookupParameterRegistry(schema::PrimitiveType_HashtableLookup, PopulateHashtableLookupParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_HashtableLookup, PopulateHashtableLookupParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/if_populate.cc b/mindspore/lite/src/ops/populate/if_populate.cc index 57b4aa8018..528a2141a3 100644 --- a/mindspore/lite/src/ops/populate/if_populate.cc +++ b/mindspore/lite/src/ops/populate/if_populate.cc @@ -15,9 +15,10 @@ */ #include "src/ops/populate/populate_register.h" #include "src/ops/populate/default_populate.h" +using mindspore::schema::PrimitiveType_If; namespace mindspore { namespace lite { -Registry g_ifParameterRegistry(schema::PrimitiveType_If, DefaultPopulateParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_If, DefaultPopulateParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/instance_norm_populate.cc b/mindspore/lite/src/ops/populate/instance_norm_populate.cc index ca4bf66b59..56aef044d7 100644 --- a/mindspore/lite/src/ops/populate/instance_norm_populate.cc +++ b/mindspore/lite/src/ops/populate/instance_norm_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/instance_norm_parameter.h" +using mindspore::schema::PrimitiveType_InstanceNorm; namespace mindspore { namespace lite { @@ -34,6 +35,6 @@ OpParameter *PopulateInstanceNormParameter(const void *prim) { return reinterpret_cast(instance_norm_param); } -Registry InstanceNormParameterRegistry(schema::PrimitiveType_InstanceNorm, PopulateInstanceNormParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_InstanceNorm, PopulateInstanceNormParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/invert_permutation_populate.cc b/mindspore/lite/src/ops/populate/invert_permutation_populate.cc index 4268621d4d..7fea6e2cdb 100644 --- a/mindspore/lite/src/ops/populate/invert_permutation_populate.cc +++ b/mindspore/lite/src/ops/populate/invert_permutation_populate.cc @@ -15,10 +15,10 @@ */ #include "src/ops/populate/populate_register.h" #include "src/ops/populate/default_populate.h" +using mindspore::schema::PrimitiveType_InvertPermutation; namespace mindspore { namespace lite { -Registry g_invertPermutationParameterRegistry(schema::PrimitiveType_InvertPermutation, DefaultPopulateParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_InvertPermutation, DefaultPopulateParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/isfinite_populate.cc b/mindspore/lite/src/ops/populate/isfinite_populate.cc index 9148ecb8b2..6d9fc4428b 100644 --- a/mindspore/lite/src/ops/populate/isfinite_populate.cc +++ b/mindspore/lite/src/ops/populate/isfinite_populate.cc @@ -15,9 +15,10 @@ */ #include "src/ops/populate/populate_register.h" #include "src/ops/populate/default_populate.h" +using mindspore::schema::PrimitiveType_IsFinite; namespace mindspore { namespace lite { -Registry g_isFiniteParameterRegistry(schema::PrimitiveType_IsFinite, DefaultPopulateParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_IsFinite, DefaultPopulateParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/l2_norm_populate.cc b/mindspore/lite/src/ops/populate/l2_norm_populate.cc index 1bfe2ef74e..f3011799dd 100644 --- a/mindspore/lite/src/ops/populate/l2_norm_populate.cc +++ b/mindspore/lite/src/ops/populate/l2_norm_populate.cc @@ -16,6 +16,7 @@ #include #include "src/ops/populate/populate_register.h" #include "nnacl/l2_norm_parameter.h" +using mindspore::schema::PrimitiveType_L2NormalizeFusion; namespace mindspore { namespace lite { @@ -53,7 +54,6 @@ OpParameter *PopulateL2NormParameter(const void *prim) { } return reinterpret_cast(l2_norm_parameter); } -Registry L2NormParameterRegistry(schema::PrimitiveType_L2NormalizeFusion, PopulateL2NormParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_L2NormalizeFusion, PopulateL2NormParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/layer_norm_grad_populate.cc b/mindspore/lite/src/ops/populate/layer_norm_grad_populate.cc index 5ea99618b8..57c37be697 100644 --- a/mindspore/lite/src/ops/populate/layer_norm_grad_populate.cc +++ b/mindspore/lite/src/ops/populate/layer_norm_grad_populate.cc @@ -16,6 +16,7 @@ #include "nnacl/fp32_grad/layernormgrad_parameter.h" #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_LayerNormGrad; namespace mindspore { namespace lite { @@ -34,7 +35,6 @@ OpParameter *PopulateLayerNormGradParameter(const void *prim) { return reinterpret_cast(layer_norm_grad_parameter); } -Registry g_layerNormGradParameterRegistry(schema::PrimitiveType_LayerNormGrad, PopulateLayerNormGradParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_LayerNormGrad, PopulateLayerNormGradParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/layer_norm_populate.cc b/mindspore/lite/src/ops/populate/layer_norm_populate.cc index b99bb95a16..24e8288798 100644 --- a/mindspore/lite/src/ops/populate/layer_norm_populate.cc +++ b/mindspore/lite/src/ops/populate/layer_norm_populate.cc @@ -16,6 +16,7 @@ #include "nnacl/layer_norm_parameter.h" #include #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_LayerNormFusion; namespace mindspore { namespace lite { OpParameter *PopulateLayerNormParameter(const void *prim) { @@ -35,6 +36,6 @@ OpParameter *PopulateLayerNormParameter(const void *prim) { return reinterpret_cast(layer_norm_parameter); } -Registry g_layerNormParameterRegistry(schema::PrimitiveType_LayerNormFusion, PopulateLayerNormParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_LayerNormFusion, PopulateLayerNormParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/lin_space_populate.cc b/mindspore/lite/src/ops/populate/lin_space_populate.cc index fdc4f1a1db..7c0b7d5f41 100644 --- a/mindspore/lite/src/ops/populate/lin_space_populate.cc +++ b/mindspore/lite/src/ops/populate/lin_space_populate.cc @@ -15,9 +15,10 @@ */ #include "src/ops/populate/populate_register.h" #include "src/ops/populate/default_populate.h" +using mindspore::schema::PrimitiveType_LinSpace; namespace mindspore { namespace lite { -Registry g_linSpaceParameterRegistry(schema::PrimitiveType_LinSpace, DefaultPopulateParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_LinSpace, DefaultPopulateParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/local_response_normalization_populate.cc b/mindspore/lite/src/ops/populate/local_response_normalization_populate.cc index a208a78a34..865ebaf8d3 100644 --- a/mindspore/lite/src/ops/populate/local_response_normalization_populate.cc +++ b/mindspore/lite/src/ops/populate/local_response_normalization_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32/local_response_norm_fp32.h" +using mindspore::schema::PrimitiveType_LRN; namespace mindspore { namespace lite { @@ -37,8 +38,7 @@ OpParameter *PopulateLocalResponseNormParameter(const void *prim) { return reinterpret_cast(lrn_param); } -Registry LocalResponseNormalizationParameterRegistry(schema::PrimitiveType_LRN, PopulateLocalResponseNormParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_LRN, PopulateLocalResponseNormParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/lsh_projection_populate.cc b/mindspore/lite/src/ops/populate/lsh_projection_populate.cc index f3b59f7aba..cdd5396172 100644 --- a/mindspore/lite/src/ops/populate/lsh_projection_populate.cc +++ b/mindspore/lite/src/ops/populate/lsh_projection_populate.cc @@ -15,6 +15,7 @@ */ #include "nnacl/lsh_projection_parameter.h" #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_LshProjection; namespace mindspore { namespace lite { @@ -34,8 +35,7 @@ OpParameter *PopulateLshProjectionParameter(const void *prim) { lsh_project_param->lsh_type_ = value->type(); return reinterpret_cast(lsh_project_param); } -Registry LshProjectionParameterRegistry(schema::PrimitiveType_LshProjection, PopulateLshProjectionParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_LshProjection, PopulateLshProjectionParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/lstm_populate.cc b/mindspore/lite/src/ops/populate/lstm_populate.cc index e94dcceeaa..54faacf05b 100644 --- a/mindspore/lite/src/ops/populate/lstm_populate.cc +++ b/mindspore/lite/src/ops/populate/lstm_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32/lstm_fp32.h" +using mindspore::schema::PrimitiveType_LSTM; namespace mindspore { namespace lite { @@ -41,6 +42,6 @@ OpParameter *PopulateLstmParameter(const void *prim) { return reinterpret_cast(lstm_param); } } // namespace -Registry g_lstmParameterRegistry(schema::PrimitiveType_LSTM, PopulateLstmParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_LSTM, PopulateLstmParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/matmul_populate.cc b/mindspore/lite/src/ops/populate/matmul_populate.cc index 8ac9314084..be0619ab93 100644 --- a/mindspore/lite/src/ops/populate/matmul_populate.cc +++ b/mindspore/lite/src/ops/populate/matmul_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/matmul_parameter.h" +using mindspore::schema::PrimitiveType_MatMul; namespace mindspore { namespace lite { @@ -34,7 +35,6 @@ OpParameter *PopulateMatMulParameter(const void *prim) { matmul_param->act_type_ = ActType_No; return reinterpret_cast(matmul_param); } -Registry MatMulParameterRegistry(schema::PrimitiveType_MatMul, PopulateMatMulParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_MatMul, PopulateMatMulParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/merge_populate.cc b/mindspore/lite/src/ops/populate/merge_populate.cc index 8ab485a46d..4fae7b67e1 100644 --- a/mindspore/lite/src/ops/populate/merge_populate.cc +++ b/mindspore/lite/src/ops/populate/merge_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_Merge; namespace mindspore { namespace lite { @@ -29,6 +30,6 @@ OpParameter *PopulateMergeParameter(const void *prim) { merge_parameter->type_ = primitive->value_type(); return reinterpret_cast(merge_parameter); } -Registry MergeParameterRegistry(schema::PrimitiveType_Merge, PopulateMergeParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Merge, PopulateMergeParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/mfcc_populate.cc b/mindspore/lite/src/ops/populate/mfcc_populate.cc index 22b2cea316..88bb64bcb1 100644 --- a/mindspore/lite/src/ops/populate/mfcc_populate.cc +++ b/mindspore/lite/src/ops/populate/mfcc_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/infer/mfcc_infer.h" +using mindspore::schema::PrimitiveType_Mfcc; namespace mindspore { namespace lite { @@ -34,6 +35,6 @@ OpParameter *PopulateMfccParameter(const void *prim) { } } // namespace -Registry g_mfccParameterRegistry(schema::PrimitiveType_Mfcc, PopulateMfccParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Mfcc, PopulateMfccParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/mul_populate.cc b/mindspore/lite/src/ops/populate/mul_populate.cc index b56957d39b..609b4b9573 100644 --- a/mindspore/lite/src/ops/populate/mul_populate.cc +++ b/mindspore/lite/src/ops/populate/mul_populate.cc @@ -16,6 +16,7 @@ #include "nnacl/arithmetic.h" #include "src/ops/populate/populate_register.h" #include "src/ops/populate/arithmetic_populate.h" +using mindspore::schema::PrimitiveType_MulFusion; namespace mindspore { namespace lite { @@ -34,6 +35,6 @@ OpParameter *PopulateMulParameter(const void *prim) { } } // namespace -Registry g_mulParameterRegistry(schema::PrimitiveType_MulFusion, PopulateMulParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_MulFusion, PopulateMulParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/non_max_suppression_populate.cc b/mindspore/lite/src/ops/populate/non_max_suppression_populate.cc index 95a49ca95f..8c0f257e9f 100644 --- a/mindspore/lite/src/ops/populate/non_max_suppression_populate.cc +++ b/mindspore/lite/src/ops/populate/non_max_suppression_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/non_max_suppression_parameter.h" +using mindspore::schema::PrimitiveType_NonMaxSuppression; namespace mindspore { namespace lite { @@ -32,8 +33,7 @@ OpParameter *PopulateNonMaxSuppressionParameter(const void *prim) { param->center_point_box_ = value->center_point_box(); return reinterpret_cast(param); } -Registry NonMaxSuppressionParameterRegistry(schema::PrimitiveType_NonMaxSuppression, PopulateNonMaxSuppressionParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_NonMaxSuppression, PopulateNonMaxSuppressionParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/nonzero_populate.cc b/mindspore/lite/src/ops/populate/nonzero_populate.cc index dfa5e2d22d..4650423d59 100644 --- a/mindspore/lite/src/ops/populate/nonzero_populate.cc +++ b/mindspore/lite/src/ops/populate/nonzero_populate.cc @@ -15,9 +15,10 @@ */ #include "src/ops/populate/populate_register.h" #include "src/ops/populate/default_populate.h" +using mindspore::schema::PrimitiveType_NonZero; namespace mindspore { namespace lite { -Registry g_nonZeroParameterRegistry(schema::PrimitiveType_NonZero, DefaultPopulateParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_NonZero, DefaultPopulateParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/one_hot_populate.cc b/mindspore/lite/src/ops/populate/one_hot_populate.cc index 1343c3d68f..becc6e1a32 100644 --- a/mindspore/lite/src/ops/populate/one_hot_populate.cc +++ b/mindspore/lite/src/ops/populate/one_hot_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32/one_hot_fp32.h" +using mindspore::schema::PrimitiveType_OneHot; namespace mindspore { namespace lite { @@ -33,7 +34,6 @@ OpParameter *PopulateOneHotParameter(const void *prim) { one_hot_param->axis_ = value->axis(); return reinterpret_cast(one_hot_param); } -Registry OneHotParameterRegistry(schema::PrimitiveType_OneHot, PopulateOneHotParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_OneHot, PopulateOneHotParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/oneslike_populate.cc b/mindspore/lite/src/ops/populate/oneslike_populate.cc index 2882d8df26..9bb3281734 100644 --- a/mindspore/lite/src/ops/populate/oneslike_populate.cc +++ b/mindspore/lite/src/ops/populate/oneslike_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_OnesLike; namespace mindspore { namespace lite { @@ -29,6 +30,6 @@ OpParameter *PopulateOnesLikeParameter(const void *prim) { return param; } -Registry OnesLikeParameterRegistry(schema::PrimitiveType_OnesLike, PopulateOnesLikeParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_OnesLike, PopulateOnesLikeParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/p_relu_populate.cc b/mindspore/lite/src/ops/populate/p_relu_populate.cc index 18a3274142..5d109d00a3 100644 --- a/mindspore/lite/src/ops/populate/p_relu_populate.cc +++ b/mindspore/lite/src/ops/populate/p_relu_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/prelu_parameter.h" +using mindspore::schema::PrimitiveType_PReLUFusion; namespace mindspore { namespace lite { @@ -32,6 +33,6 @@ OpParameter *PopulatePReLUParameter(const void *prim) { param->channelShared = value->channel_shared(); return reinterpret_cast(param); } -Registry PReLUParameterRegistry(schema::PrimitiveType_PReLUFusion, PopulatePReLUParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_PReLUFusion, PopulatePReLUParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/pad_populate.cc b/mindspore/lite/src/ops/populate/pad_populate.cc index 58a3945711..ae5c16a1e4 100644 --- a/mindspore/lite/src/ops/populate/pad_populate.cc +++ b/mindspore/lite/src/ops/populate/pad_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/pad_parameter.h" +using mindspore::schema::PrimitiveType_PadFusion; namespace mindspore { namespace lite { @@ -33,7 +34,6 @@ OpParameter *PopulatePadParameter(const void *prim) { pad_param->constant_value_ = value->constant_value(); return reinterpret_cast(pad_param); } -Registry PadParameterRegistry(schema::PrimitiveType_PadFusion, PopulatePadParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_PadFusion, PopulatePadParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/partial_populate.cc b/mindspore/lite/src/ops/populate/partial_populate.cc index f808a530af..4ff60edffb 100644 --- a/mindspore/lite/src/ops/populate/partial_populate.cc +++ b/mindspore/lite/src/ops/populate/partial_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_PartialFusion; namespace mindspore { namespace lite { @@ -36,6 +37,6 @@ OpParameter *PopulatePartialParameter(const void *prim) { return reinterpret_cast(partial_parameter); } -Registry PartialParameterRegistry(schema::PrimitiveType_PartialFusion, PopulatePartialParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_PartialFusion, PopulatePartialParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/pooling_populate.cc b/mindspore/lite/src/ops/populate/pooling_populate.cc index af368282bb..6eeabe289a 100644 --- a/mindspore/lite/src/ops/populate/pooling_populate.cc +++ b/mindspore/lite/src/ops/populate/pooling_populate.cc @@ -15,6 +15,8 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/pooling_parameter.h" +using mindspore::schema::PrimitiveType_AvgPoolFusion; +using mindspore::schema::PrimitiveType_MaxPoolFusion; namespace mindspore { namespace lite { @@ -140,7 +142,7 @@ OpParameter *PopulateMaxPoolParameter(const void *primitive) { } } // namespace -Registry g_avgPoolParameterRegistry(schema::PrimitiveType_AvgPoolFusion, PopulateAvgPoolParameter, SCHEMA_CUR); -Registry g_maxPoolParameterRegistry(schema::PrimitiveType_MaxPoolFusion, PopulateMaxPoolParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_AvgPoolFusion, PopulateAvgPoolParameter, SCHEMA_CUR) +REG_POPULATE(PrimitiveType_MaxPoolFusion, PopulateMaxPoolParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/power_populate.cc b/mindspore/lite/src/ops/populate/power_populate.cc index 26b5bfdcc9..1a0cc30087 100644 --- a/mindspore/lite/src/ops/populate/power_populate.cc +++ b/mindspore/lite/src/ops/populate/power_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/power_parameter.h" +using mindspore::schema::PrimitiveType_PowFusion; namespace mindspore { namespace lite { @@ -35,6 +36,6 @@ OpParameter *PopulatePowerParameter(const void *prim) { } } // namespace -Registry g_powerParameterRegistry(schema::PrimitiveType_PowFusion, PopulatePowerParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_PowFusion, PopulatePowerParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/prior_box_populate.cc b/mindspore/lite/src/ops/populate/prior_box_populate.cc index 611b594dcf..0382499a87 100644 --- a/mindspore/lite/src/ops/populate/prior_box_populate.cc +++ b/mindspore/lite/src/ops/populate/prior_box_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/prior_box_parameter.h" +using mindspore::schema::PrimitiveType_PriorBox; namespace mindspore { namespace lite { @@ -69,7 +70,6 @@ OpParameter *PopulatePriorBoxParameter(const void *prim) { prior_box_param->step_w = value->step_w(); return reinterpret_cast(prior_box_param); } -Registry PriorBoxParameterRegistry(schema::PrimitiveType_PriorBox, PopulatePriorBoxParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_PriorBox, PopulatePriorBoxParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/quant_dtype_cast_populate.cc b/mindspore/lite/src/ops/populate/quant_dtype_cast_populate.cc index d49640110a..251e48f7b1 100644 --- a/mindspore/lite/src/ops/populate/quant_dtype_cast_populate.cc +++ b/mindspore/lite/src/ops/populate/quant_dtype_cast_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/int8/quant_dtype_cast_int8.h" +using mindspore::schema::PrimitiveType_QuantDTypeCast; namespace mindspore { namespace lite { @@ -34,8 +35,7 @@ OpParameter *PopulateQuantDTypeCastParameter(const void *prim) { parameter->dstT = value->dst_t(); return reinterpret_cast(parameter); } -Registry QuantDTypeCastParameterRegistry(schema::PrimitiveType_QuantDTypeCast, PopulateQuantDTypeCastParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_QuantDTypeCast, PopulateQuantDTypeCastParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/random_standard_normal_populate.cc b/mindspore/lite/src/ops/populate/random_standard_normal_populate.cc index bbc0a2ebda..d19fd37749 100644 --- a/mindspore/lite/src/ops/populate/random_standard_normal_populate.cc +++ b/mindspore/lite/src/ops/populate/random_standard_normal_populate.cc @@ -16,6 +16,7 @@ #include "src/ops/populate/populate_register.h" #include "nnacl/random_parameter.h" +using mindspore::schema::PrimitiveType_RandomStandardNormal; namespace mindspore { namespace lite { @@ -35,7 +36,6 @@ OpParameter *PopulateRandomStandardNormalParameter(const void *prim) { return reinterpret_cast(random_parameter); } } // namespace -Registry g_randomStandardNormalParameterRegistry(schema::PrimitiveType_RandomStandardNormal, - PopulateRandomStandardNormalParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_RandomStandardNormal, PopulateRandomStandardNormalParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/range_populate.cc b/mindspore/lite/src/ops/populate/range_populate.cc index 2f30e10d92..8fc506a294 100644 --- a/mindspore/lite/src/ops/populate/range_populate.cc +++ b/mindspore/lite/src/ops/populate/range_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32/range_fp32.h" +using mindspore::schema::PrimitiveType_Range; namespace mindspore { namespace lite { @@ -36,7 +37,6 @@ OpParameter *PopulateRangeParameter(const void *prim) { return reinterpret_cast(range_param); } } // namespace -Registry g_rangeParameterRegistry(schema::PrimitiveType_Range, PopulateRangeParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_Range, PopulateRangeParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/rank_populate.cc b/mindspore/lite/src/ops/populate/rank_populate.cc index f9dd3373fa..b239e77a55 100644 --- a/mindspore/lite/src/ops/populate/rank_populate.cc +++ b/mindspore/lite/src/ops/populate/rank_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_Rank; namespace mindspore { namespace lite { @@ -31,6 +32,6 @@ OpParameter *PopulateRankParameter(const void *prim) { } } // namespace -Registry g_rankParameterRegistry(schema::PrimitiveType_Rank, PopulateRankParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Rank, PopulateRankParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/reshape_populate.cc b/mindspore/lite/src/ops/populate/reshape_populate.cc index a4685326d1..ccf9894626 100644 --- a/mindspore/lite/src/ops/populate/reshape_populate.cc +++ b/mindspore/lite/src/ops/populate/reshape_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/reshape_parameter.h" +using mindspore::schema::PrimitiveType_Reshape; namespace mindspore { namespace lite { @@ -32,6 +33,6 @@ OpParameter *PopulateReshapeParameter(const void *prim) { } } // namespace -Registry g_reshapeParameterRegistry(schema::PrimitiveType_Reshape, PopulateReshapeParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Reshape, PopulateReshapeParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/resize_populate.cc b/mindspore/lite/src/ops/populate/resize_populate.cc index c0a4c6f002..32b3882025 100644 --- a/mindspore/lite/src/ops/populate/resize_populate.cc +++ b/mindspore/lite/src/ops/populate/resize_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/resize_parameter.h" +using mindspore::schema::PrimitiveType_Resize; namespace mindspore { namespace lite { @@ -39,7 +40,7 @@ OpParameter *PopulateResizeParameter(const void *prim) { return reinterpret_cast(resize_param); } -Registry ResizeParameterRegistry(schema::PrimitiveType_Resize, PopulateResizeParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Resize, PopulateResizeParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/reverse_populate.cc b/mindspore/lite/src/ops/populate/reverse_populate.cc index 04beec67fa..281e89e8ca 100644 --- a/mindspore/lite/src/ops/populate/reverse_populate.cc +++ b/mindspore/lite/src/ops/populate/reverse_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32/reverse_fp32.h" +using mindspore::schema::PrimitiveType_ReverseV2; namespace mindspore { namespace lite { @@ -39,7 +40,6 @@ OpParameter *PopulateReverseParameter(const void *prim) { return reinterpret_cast(reverse_param); } -Registry ReverseParameterRegistry(schema::PrimitiveType_ReverseV2, PopulateReverseParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_ReverseV2, PopulateReverseParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/reverse_sequence_populate.cc b/mindspore/lite/src/ops/populate/reverse_sequence_populate.cc index 86eaa35c05..f5b26df147 100644 --- a/mindspore/lite/src/ops/populate/reverse_sequence_populate.cc +++ b/mindspore/lite/src/ops/populate/reverse_sequence_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/reverse_sequence_parameter.h" +using mindspore::schema::PrimitiveType_ReverseSequence; namespace mindspore { namespace lite { @@ -36,8 +37,7 @@ OpParameter *PopulateReverseSequenceParameter(const void *prim) { } } // namespace -Registry ReverseSequenceParameterRegistry(schema::PrimitiveType_ReverseSequence, PopulateReverseSequenceParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_ReverseSequence, PopulateReverseSequenceParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/roi_pooling_populate.cc b/mindspore/lite/src/ops/populate/roi_pooling_populate.cc index 5867a0701f..9e9baf9b19 100644 --- a/mindspore/lite/src/ops/populate/roi_pooling_populate.cc +++ b/mindspore/lite/src/ops/populate/roi_pooling_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32/roi_pooling_fp32.h" +using mindspore::schema::PrimitiveType_ROIPooling; namespace mindspore { namespace lite { @@ -37,6 +38,6 @@ OpParameter *PopulateROIPoolingParameter(const void *prim) { } } // namespace -Registry g_ROIPoolingParameterRegistry(schema::PrimitiveType_ROIPooling, PopulateROIPoolingParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_ROIPooling, PopulateROIPoolingParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/scale_populate.cc b/mindspore/lite/src/ops/populate/scale_populate.cc index 923fbcbf8f..5a5de2d721 100644 --- a/mindspore/lite/src/ops/populate/scale_populate.cc +++ b/mindspore/lite/src/ops/populate/scale_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/scale.h" +using mindspore::schema::PrimitiveType_ScaleFusion; namespace mindspore { namespace lite { @@ -35,6 +36,6 @@ OpParameter *PopulateScaleParameter(const void *prim) { } } // namespace -Registry g_scaleParameterRegistry(schema::PrimitiveType_ScaleFusion, PopulateScaleParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_ScaleFusion, PopulateScaleParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/scatter_nd_populate.cc b/mindspore/lite/src/ops/populate/scatter_nd_populate.cc index 4a8dfa4b17..2ef2a1b4dc 100644 --- a/mindspore/lite/src/ops/populate/scatter_nd_populate.cc +++ b/mindspore/lite/src/ops/populate/scatter_nd_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_ScatterNd; namespace mindspore { namespace lite { @@ -30,7 +31,6 @@ OpParameter *PopulateScatterNDParameter(const void *prim) { return reinterpret_cast(scatter_nd_param); } } // namespace -Registry g_scatterNDParameterRegistry(schema::PrimitiveType_ScatterNd, PopulateScatterNDParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_ScatterNd, PopulateScatterNDParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/select_populate.cc b/mindspore/lite/src/ops/populate/select_populate.cc index 2ed6cd9d2d..eb3e79812a 100644 --- a/mindspore/lite/src/ops/populate/select_populate.cc +++ b/mindspore/lite/src/ops/populate/select_populate.cc @@ -15,9 +15,10 @@ */ #include "src/ops/populate/populate_register.h" #include "src/ops/populate/default_populate.h" +using mindspore::schema::PrimitiveType_Select; namespace mindspore { namespace lite { -Registry g_selectParameterRegistry(schema::PrimitiveType_Select, DefaultPopulateParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Select, DefaultPopulateParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/shape_populate.cc b/mindspore/lite/src/ops/populate/shape_populate.cc index eab41e1144..4dc4ee0854 100644 --- a/mindspore/lite/src/ops/populate/shape_populate.cc +++ b/mindspore/lite/src/ops/populate/shape_populate.cc @@ -16,6 +16,7 @@ #include "src/ops/populate/populate_register.h" #include "src/common/log_adapter.h" #include "src/tensor.h" +using mindspore::schema::PrimitiveType_Shape; namespace mindspore { namespace lite { @@ -33,6 +34,6 @@ OpParameter *PopulateShapeParameter(const void *prim) { } } // namespace -Registry g_shapeParameterRegistry(schema::PrimitiveType_Shape, PopulateShapeParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Shape, PopulateShapeParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/size_populate.cc b/mindspore/lite/src/ops/populate/size_populate.cc index 7c35d9e7c4..3b3516b3b4 100644 --- a/mindspore/lite/src/ops/populate/size_populate.cc +++ b/mindspore/lite/src/ops/populate/size_populate.cc @@ -15,9 +15,10 @@ */ #include "src/ops/populate/populate_register.h" #include "src/ops/populate/default_populate.h" +using mindspore::schema::PrimitiveType_Size; namespace mindspore { namespace lite { -Registry g_sizeParameterRegistry(schema::PrimitiveType_Size, DefaultPopulateParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Size, DefaultPopulateParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/skip_gram_populate.cc b/mindspore/lite/src/ops/populate/skip_gram_populate.cc index bfe59511d9..54324b8367 100644 --- a/mindspore/lite/src/ops/populate/skip_gram_populate.cc +++ b/mindspore/lite/src/ops/populate/skip_gram_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/skip_gram_parameter.h" +using mindspore::schema::PrimitiveType_SkipGram; namespace mindspore { namespace lite { @@ -34,7 +35,6 @@ OpParameter *PopulateSkipGramParameter(const void *prim) { skipGramParameter->include_all_ngrams = value->include_all_grams(); return reinterpret_cast(skipGramParameter); } -Registry SkipGramParameterRegistry(schema::PrimitiveType_SkipGram, PopulateSkipGramParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_SkipGram, PopulateSkipGramParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/slice_populate.cc b/mindspore/lite/src/ops/populate/slice_populate.cc index fa224d2cd4..3003ab6162 100644 --- a/mindspore/lite/src/ops/populate/slice_populate.cc +++ b/mindspore/lite/src/ops/populate/slice_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/slice_parameter.h" +using mindspore::schema::PrimitiveType_SliceFusion; namespace mindspore { namespace lite { @@ -34,7 +35,6 @@ OpParameter *PopulateSliceParameter(const void *prim) { } return reinterpret_cast(slice_param); } -Registry SliceParameterRegistry(schema::PrimitiveType_SliceFusion, PopulateSliceParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_SliceFusion, PopulateSliceParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/softmax_populate.cc b/mindspore/lite/src/ops/populate/softmax_populate.cc index 11ee873f10..e05d872c58 100644 --- a/mindspore/lite/src/ops/populate/softmax_populate.cc +++ b/mindspore/lite/src/ops/populate/softmax_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/softmax_parameter.h" +using mindspore::schema::PrimitiveType_Softmax; namespace mindspore { namespace lite { @@ -39,6 +40,6 @@ OpParameter *PopulateSoftmaxParameter(const void *prim) { } } // namespace -Registry g_softmaxParameterRegistry(schema::PrimitiveType_Softmax, PopulateSoftmaxParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Softmax, PopulateSoftmaxParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/space_to_batch_nd_populate.cc b/mindspore/lite/src/ops/populate/space_to_batch_nd_populate.cc index af71de033f..e740f235e3 100644 --- a/mindspore/lite/src/ops/populate/space_to_batch_nd_populate.cc +++ b/mindspore/lite/src/ops/populate/space_to_batch_nd_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32/space_to_batch_fp32.h" +using mindspore::schema::PrimitiveType_SpaceToBatchND; namespace mindspore { namespace lite { @@ -67,6 +68,6 @@ OpParameter *PopulateSpaceToBatchNDParameter(const void *prim) { return reinterpret_cast(space_batch_param_nd); } } // namespace -Registry g_spaceToBatchNDRegistry(schema::PrimitiveType_SpaceToBatchND, PopulateSpaceToBatchNDParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_SpaceToBatchND, PopulateSpaceToBatchNDParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/space_to_batch_populate.cc b/mindspore/lite/src/ops/populate/space_to_batch_populate.cc index 75077c72bd..c9ae8875c8 100644 --- a/mindspore/lite/src/ops/populate/space_to_batch_populate.cc +++ b/mindspore/lite/src/ops/populate/space_to_batch_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32/space_to_batch_fp32.h" +using mindspore::schema::PrimitiveType_SpaceToBatch; namespace mindspore { namespace lite { @@ -63,6 +64,6 @@ OpParameter *PopulateSpaceToBatchParameter(const void *prim) { return reinterpret_cast(space_batch_param); } } // namespace -Registry g_spaceToBatchRegistry(schema::PrimitiveType_SpaceToBatch, PopulateSpaceToBatchParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_SpaceToBatch, PopulateSpaceToBatchParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/space_to_depth_populate.cc b/mindspore/lite/src/ops/populate/space_to_depth_populate.cc index c78e23382d..270d8fc67e 100644 --- a/mindspore/lite/src/ops/populate/space_to_depth_populate.cc +++ b/mindspore/lite/src/ops/populate/space_to_depth_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/space_to_depth_parameter.h" +using mindspore::schema::PrimitiveType_SpaceToDepth; namespace mindspore { namespace lite { @@ -37,6 +38,6 @@ OpParameter *PopulateSpaceToDepthParameter(const void *prim) { } return reinterpret_cast(space_depth_param); } -Registry SpaceToDepthParameterRegistry(schema::PrimitiveType_SpaceToDepth, PopulateSpaceToDepthParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_SpaceToDepth, PopulateSpaceToDepthParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/sparse_softmax_cross_entropy_with_logits.cc b/mindspore/lite/src/ops/populate/sparse_softmax_cross_entropy_with_logits.cc index 4b70fc5e2b..7a9541202c 100644 --- a/mindspore/lite/src/ops/populate/sparse_softmax_cross_entropy_with_logits.cc +++ b/mindspore/lite/src/ops/populate/sparse_softmax_cross_entropy_with_logits.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32_grad/softmax_grad.h" +using mindspore::schema::PrimitiveType_SparseSoftmaxCrossEntropyWithLogits; namespace mindspore { namespace lite { @@ -30,8 +31,7 @@ OpParameter *PopulateSparseSoftmaxCrossEntropyWithLogitsParameter(const void *pr softmax_cross_entropy_param_->op_parameter_.type_ = primitive->value_type(); return reinterpret_cast(softmax_cross_entropy_param_); } -Registry SparseSoftmaxCrossEntropyWithLogitsParameterRegistry(schema::PrimitiveType_SparseSoftmaxCrossEntropyWithLogits, - PopulateSparseSoftmaxCrossEntropyWithLogitsParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_SparseSoftmaxCrossEntropyWithLogits, PopulateSparseSoftmaxCrossEntropyWithLogitsParameter, + SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/sparse_to_dense_populate.cc b/mindspore/lite/src/ops/populate/sparse_to_dense_populate.cc index 578824279d..87fc2b6bd6 100644 --- a/mindspore/lite/src/ops/populate/sparse_to_dense_populate.cc +++ b/mindspore/lite/src/ops/populate/sparse_to_dense_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/sparse_to_dense_parameter.h" +using mindspore::schema::PrimitiveType_SparseToDense; namespace mindspore { namespace lite { @@ -32,7 +33,6 @@ OpParameter *PopulateSparseToDenseParameter(const void *prim) { } } // namespace -Registry g_sparseToDenseParameterRegistry(schema::PrimitiveType_SparseToDense, PopulateSparseToDenseParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_SparseToDense, PopulateSparseToDenseParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/splice_populate.cc b/mindspore/lite/src/ops/populate/splice_populate.cc index 6b74e3ef53..6563697864 100644 --- a/mindspore/lite/src/ops/populate/splice_populate.cc +++ b/mindspore/lite/src/ops/populate/splice_populate.cc @@ -16,6 +16,7 @@ #include "src/ops/populate/populate_register.h" #include "nnacl/op_base.h" #include "nnacl/splice_parameter.h" +using mindspore::schema::PrimitiveType_Splice; namespace mindspore { namespace lite { OpParameter *PopulateSpliceParameter(const void *prim) { @@ -67,6 +68,6 @@ OpParameter *PopulateSpliceParameter(const void *prim) { splice_parameter->output_dim_ = splice_primitive->output_dim(); return reinterpret_cast(splice_parameter); } -Registry g_SpliceParameterRegistry(schema::PrimitiveType_Splice, PopulateSpliceParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Splice, PopulateSpliceParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/split_populate.cc b/mindspore/lite/src/ops/populate/split_populate.cc index 74c32b024d..270cbada0e 100644 --- a/mindspore/lite/src/ops/populate/split_populate.cc +++ b/mindspore/lite/src/ops/populate/split_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/split_parameter.h" +using mindspore::schema::PrimitiveType_Split; namespace mindspore { namespace lite { @@ -59,6 +60,6 @@ OpParameter *PopulateSplitParameter(const void *prim) { return reinterpret_cast(split_param); } } // namespace -Registry g_splitParameterRegistry(schema::PrimitiveType_Split, PopulateSplitParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Split, PopulateSplitParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/squeeze_populate.cc b/mindspore/lite/src/ops/populate/squeeze_populate.cc index 0b0e91bc69..e29dbe39ec 100644 --- a/mindspore/lite/src/ops/populate/squeeze_populate.cc +++ b/mindspore/lite/src/ops/populate/squeeze_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/squeeze_parameter.h" +using mindspore::schema::PrimitiveType_Squeeze; namespace mindspore { namespace lite { @@ -42,6 +43,6 @@ OpParameter *PopulateSqueezeParameter(const void *prim) { return reinterpret_cast(squeeze_param); } } // namespace -Registry g_squeezeParameterRegistry(schema::PrimitiveType_Squeeze, PopulateSqueezeParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Squeeze, PopulateSqueezeParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/stack_populate.cc b/mindspore/lite/src/ops/populate/stack_populate.cc index 7eede89858..ca4b375f4b 100644 --- a/mindspore/lite/src/ops/populate/stack_populate.cc +++ b/mindspore/lite/src/ops/populate/stack_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/stack_parameter.h" +using mindspore::schema::PrimitiveType_Stack; namespace mindspore { namespace lite { @@ -33,6 +34,6 @@ OpParameter *PopulateStackParameter(const void *prim) { return reinterpret_cast(stack_param); } } // namespace -Registry g_stackParameterRegistry(schema::PrimitiveType_Stack, PopulateStackParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Stack, PopulateStackParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/strided_slice_grad_populate.cc b/mindspore/lite/src/ops/populate/strided_slice_grad_populate.cc index a047d3249c..b8e2b534f1 100644 --- a/mindspore/lite/src/ops/populate/strided_slice_grad_populate.cc +++ b/mindspore/lite/src/ops/populate/strided_slice_grad_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/strided_slice_parameter.h" +using mindspore::schema::PrimitiveType_StridedSliceGrad; namespace mindspore { namespace lite { @@ -38,8 +39,7 @@ OpParameter *PopulateStridedSliceGradParameter(const void *prim) { return reinterpret_cast(strided_slice_param); } -Registry StridedSliceGradParameterRegistry(schema::PrimitiveType_StridedSliceGrad, PopulateStridedSliceGradParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_StridedSliceGrad, PopulateStridedSliceGradParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/strided_slice_populate.cc b/mindspore/lite/src/ops/populate/strided_slice_populate.cc index 894a6b121e..32087bd4d1 100644 --- a/mindspore/lite/src/ops/populate/strided_slice_populate.cc +++ b/mindspore/lite/src/ops/populate/strided_slice_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/strided_slice_populate.h" +using mindspore::schema::PrimitiveType_StridedSlice; namespace mindspore { namespace lite { @@ -38,7 +39,6 @@ OpParameter *PopulateStridedSliceParameter(const void *prim) { return reinterpret_cast(strided_slice_param); } -Registry StridedSliceParameterRegistry(schema::PrimitiveType_StridedSlice, PopulateStridedSliceParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_StridedSlice, PopulateStridedSliceParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/sub_populate.cc b/mindspore/lite/src/ops/populate/sub_populate.cc index 78c8d5186d..0ce901da18 100644 --- a/mindspore/lite/src/ops/populate/sub_populate.cc +++ b/mindspore/lite/src/ops/populate/sub_populate.cc @@ -16,6 +16,7 @@ #include "src/ops/populate/populate_register.h" #include "nnacl/arithmetic.h" #include "src/ops/populate/arithmetic_populate.h" +using mindspore::schema::PrimitiveType_SubFusion; namespace mindspore { namespace lite { @@ -31,7 +32,6 @@ OpParameter *PopulateSubParameter(const void *prim) { return reinterpret_cast(param); } -Registry g_subParameterRegistry(schema::PrimitiveType_SubFusion, PopulateSubParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_SubFusion, PopulateSubParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/switch_populate.cc b/mindspore/lite/src/ops/populate/switch_populate.cc index b06e0de518..4859470fbd 100644 --- a/mindspore/lite/src/ops/populate/switch_populate.cc +++ b/mindspore/lite/src/ops/populate/switch_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_Switch; namespace mindspore { namespace lite { @@ -29,6 +30,6 @@ OpParameter *PopulateSwitchParameter(const void *prim) { return reinterpret_cast(switch_parameter); } -Registry SwitchParameterRegistry(schema::PrimitiveType_Switch, PopulateSwitchParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Switch, PopulateSwitchParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/tensorlistfromtensor_populate.cc b/mindspore/lite/src/ops/populate/tensorlistfromtensor_populate.cc index f80b76263b..d88f504d01 100644 --- a/mindspore/lite/src/ops/populate/tensorlistfromtensor_populate.cc +++ b/mindspore/lite/src/ops/populate/tensorlistfromtensor_populate.cc @@ -16,6 +16,7 @@ #include "nnacl/tensorlist_parameter.h" #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_TensorListFromTensor; namespace mindspore { namespace lite { @@ -33,8 +34,7 @@ OpParameter *PopulateTensorListFromTensorParameter(const void *prim) { TensorList_param->element_dtype_ = value->element_dtype(); return reinterpret_cast(TensorList_param); } -Registry TensorListFromTensorParameterRegistry(schema::PrimitiveType_TensorListFromTensor, - PopulateTensorListFromTensorParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_TensorListFromTensor, PopulateTensorListFromTensorParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/tensorlistgetitem_populate.cc b/mindspore/lite/src/ops/populate/tensorlistgetitem_populate.cc index 40db0cde97..a30fb5c3c2 100644 --- a/mindspore/lite/src/ops/populate/tensorlistgetitem_populate.cc +++ b/mindspore/lite/src/ops/populate/tensorlistgetitem_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/tensorlist_parameter.h" +using mindspore::schema::PrimitiveType_TensorListGetItem; namespace mindspore { namespace lite { @@ -31,8 +32,7 @@ OpParameter *PopulateTensorListGetItemParameter(const void *prim) { getItem_param->element_dtype_ = value->element_dtype(); return reinterpret_cast(getItem_param); } -Registry TensorListGetItemParameterRegistry(schema::PrimitiveType_TensorListGetItem, PopulateTensorListGetItemParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_TensorListGetItem, PopulateTensorListGetItemParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/tensorlistreserve_populate.cc b/mindspore/lite/src/ops/populate/tensorlistreserve_populate.cc index dadcb9f799..0e5cba5e5e 100644 --- a/mindspore/lite/src/ops/populate/tensorlistreserve_populate.cc +++ b/mindspore/lite/src/ops/populate/tensorlistreserve_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/tensorlist_parameter.h" +using mindspore::schema::PrimitiveType_TensorListReserve; namespace mindspore { namespace lite { @@ -31,8 +32,7 @@ OpParameter *PopulateTensorListReserveParameter(const void *prim) { reserve_param->element_dtype_ = value->element_dtype(); return reinterpret_cast(reserve_param); } -Registry TensorListReserveParameterRegistry(schema::PrimitiveType_TensorListReserve, PopulateTensorListReserveParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_TensorListReserve, PopulateTensorListReserveParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/tensorlistsetlitem_populate.cc b/mindspore/lite/src/ops/populate/tensorlistsetlitem_populate.cc index 36a0788b29..9ca3169a4f 100644 --- a/mindspore/lite/src/ops/populate/tensorlistsetlitem_populate.cc +++ b/mindspore/lite/src/ops/populate/tensorlistsetlitem_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/tensorlist_parameter.h" +using mindspore::schema::PrimitiveType_TensorListSetItem; namespace mindspore { namespace lite { @@ -31,8 +32,7 @@ OpParameter *PopulateTensorListSetItemParameter(const void *prim) { setItem_param->element_dtype_ = value->element_dtype(); return reinterpret_cast(setItem_param); } -Registry TensorListSetItemParameterRegistry(schema::PrimitiveType_TensorListSetItem, PopulateTensorListSetItemParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_TensorListSetItem, PopulateTensorListSetItemParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/tensorliststack_populate.cc b/mindspore/lite/src/ops/populate/tensorliststack_populate.cc index 615c142a75..f955099630 100644 --- a/mindspore/lite/src/ops/populate/tensorliststack_populate.cc +++ b/mindspore/lite/src/ops/populate/tensorliststack_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/tensorlist_parameter.h" +using mindspore::schema::PrimitiveType_TensorListStack; namespace mindspore { namespace lite { @@ -32,8 +33,7 @@ OpParameter *PopulateTensorListStackParameter(const void *prim) { stack_param->num_element_ = value->num_elements(); return reinterpret_cast(stack_param); } -Registry TensorListStackParameterRegistry(schema::PrimitiveType_TensorListStack, PopulateTensorListStackParameter, - SCHEMA_CUR); +REG_POPULATE(PrimitiveType_TensorListStack, PopulateTensorListStackParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/tile_populate.cc b/mindspore/lite/src/ops/populate/tile_populate.cc index 73eafa0b6a..66bfee8ca4 100644 --- a/mindspore/lite/src/ops/populate/tile_populate.cc +++ b/mindspore/lite/src/ops/populate/tile_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/base/tile_base.h" +using mindspore::schema::PrimitiveType_TileFusion; namespace mindspore { namespace lite { @@ -39,7 +40,6 @@ OpParameter *PopulateTileParameter(const void *prim) { return reinterpret_cast(tile_param); } -Registry TileParameterRegistry(schema::PrimitiveType_TileFusion, PopulateTileParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_TileFusion, PopulateTileParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/topk_populate.cc b/mindspore/lite/src/ops/populate/topk_populate.cc index 17b03004c8..e2dc4a0ee2 100644 --- a/mindspore/lite/src/ops/populate/topk_populate.cc +++ b/mindspore/lite/src/ops/populate/topk_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32/topk_fp32.h" +using mindspore::schema::PrimitiveType_TopKFusion; namespace mindspore { namespace lite { @@ -33,7 +34,6 @@ OpParameter *PopulateTopKParameter(const void *prim) { return reinterpret_cast(topk_param); } } // namespace -Registry g_topKParameterRegistry(schema::PrimitiveType_TopKFusion, PopulateTopKParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_TopKFusion, PopulateTopKParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/transpose_populate.cc b/mindspore/lite/src/ops/populate/transpose_populate.cc index 8c647dbddb..53aa90ecca 100644 --- a/mindspore/lite/src/ops/populate/transpose_populate.cc +++ b/mindspore/lite/src/ops/populate/transpose_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/transpose.h" +using mindspore::schema::PrimitiveType_Transpose; namespace mindspore { namespace lite { @@ -32,7 +33,6 @@ OpParameter *PopulateTransposeParameter(const void *prim) { } } // namespace -Registry g_transposeParameterRegistry(schema::PrimitiveType_Transpose, PopulateTransposeParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_Transpose, PopulateTransposeParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/uniform_real_populate.cc b/mindspore/lite/src/ops/populate/uniform_real_populate.cc index 37c28946ff..c3123106b0 100644 --- a/mindspore/lite/src/ops/populate/uniform_real_populate.cc +++ b/mindspore/lite/src/ops/populate/uniform_real_populate.cc @@ -16,6 +16,7 @@ #include "src/ops/populate/populate_register.h" #include "src/ops/populate/default_populate.h" #include "nnacl/random_parameter.h" +using mindspore::schema::PrimitiveType_UniformReal; namespace mindspore { namespace lite { @@ -34,6 +35,6 @@ OpParameter *PopulateRandomStandardNormalParameter(const void *prim) { return reinterpret_cast(random_parameter); } -Registry g_uniformRealParameterRegistry(schema::PrimitiveType_UniformReal, DefaultPopulateParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_UniformReal, DefaultPopulateParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/unique_populate.cc b/mindspore/lite/src/ops/populate/unique_populate.cc index abc028d76b..a5ae402a31 100644 --- a/mindspore/lite/src/ops/populate/unique_populate.cc +++ b/mindspore/lite/src/ops/populate/unique_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/fp32/unique_fp32.h" +using mindspore::schema::PrimitiveType_Unique; namespace mindspore { namespace lite { @@ -32,7 +33,6 @@ OpParameter *PopulateUniqueParameter(const void *prim) { } } // namespace -Registry g_uniqueParameterRegistry(schema::PrimitiveType_Unique, PopulateUniqueParameter, SCHEMA_CUR); - +REG_POPULATE(PrimitiveType_Unique, PopulateUniqueParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/unsorted_segment_sum_populate.cc b/mindspore/lite/src/ops/populate/unsorted_segment_sum_populate.cc index 1066a3baab..fdec1f31d0 100644 --- a/mindspore/lite/src/ops/populate/unsorted_segment_sum_populate.cc +++ b/mindspore/lite/src/ops/populate/unsorted_segment_sum_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_UnsortedSegmentSum; namespace mindspore { namespace lite { @@ -29,7 +30,6 @@ OpParameter *PopulateUnsortedSegmentSumParameter(const void *prim) { return param; } -Registry UnsortedSegmentSumParameterRegistry(schema::PrimitiveType_UnsortedSegmentSum, - PopulateUnsortedSegmentSumParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_UnsortedSegmentSum, PopulateUnsortedSegmentSumParameter, SCHEMA_CUR); } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/unsqueeze_populate.cc b/mindspore/lite/src/ops/populate/unsqueeze_populate.cc index aefa55b564..ea256a221e 100644 --- a/mindspore/lite/src/ops/populate/unsqueeze_populate.cc +++ b/mindspore/lite/src/ops/populate/unsqueeze_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/unsqueeze_parameter.h" +using mindspore::schema::PrimitiveType_Unsqueeze; namespace mindspore { namespace lite { @@ -39,6 +40,6 @@ OpParameter *PopulateUnsqueezeParameter(const void *prim) { } } // namespace -Registry g_unsqueezeParameterRegistry(schema::PrimitiveType_Unsqueeze, PopulateUnsqueezeParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Unsqueeze, PopulateUnsqueezeParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/unstack_populate.cc b/mindspore/lite/src/ops/populate/unstack_populate.cc index ab2bc59295..1109697fc2 100644 --- a/mindspore/lite/src/ops/populate/unstack_populate.cc +++ b/mindspore/lite/src/ops/populate/unstack_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/unstack_parameter.h" +using mindspore::schema::PrimitiveType_Unstack; namespace mindspore { namespace lite { @@ -32,6 +33,6 @@ OpParameter *PopulateUnstackParameter(const void *prim) { unstack_param->axis_ = value->axis(); return reinterpret_cast(unstack_param); } -Registry UnstackParameterRegistry(schema::PrimitiveType_Unstack, PopulateUnstackParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Unstack, PopulateUnstackParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/where_populate.cc b/mindspore/lite/src/ops/populate/where_populate.cc index 75f624efb6..2aed0429bf 100644 --- a/mindspore/lite/src/ops/populate/where_populate.cc +++ b/mindspore/lite/src/ops/populate/where_populate.cc @@ -15,6 +15,7 @@ */ #include "src/ops/populate/populate_register.h" #include "nnacl/where_parameter.h" +using mindspore::schema::PrimitiveType_Where; namespace mindspore { namespace lite { @@ -31,6 +32,6 @@ OpParameter *PopulateWhereParameter(const void *prim) { return reinterpret_cast(where_parameter); } } // namespace -Registry g_whereParameterRegistry(schema::PrimitiveType_Where, PopulateWhereParameter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_Where, PopulateWhereParameter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/ops/populate/while_populate.cc b/mindspore/lite/src/ops/populate/while_populate.cc index f040731015..7756ecea8a 100644 --- a/mindspore/lite/src/ops/populate/while_populate.cc +++ b/mindspore/lite/src/ops/populate/while_populate.cc @@ -14,6 +14,7 @@ * limitations under the License. */ #include "src/ops/populate/populate_register.h" +using mindspore::schema::PrimitiveType_While; namespace mindspore { namespace lite { @@ -38,6 +39,6 @@ OpParameter *PopulateWhileParemeter(const void *prim) { while_paremeter->cond_subgraph_index = value->cond_subgraph_index(); return reinterpret_cast(while_paremeter); } -Registry WhileParemeterRegistry(schema::PrimitiveType_While, PopulateWhileParemeter, SCHEMA_CUR); +REG_POPULATE(PrimitiveType_While, PopulateWhileParemeter, SCHEMA_CUR) } // namespace lite } // namespace mindspore diff --git a/mindspore/lite/src/reg_kernels.h b/mindspore/lite/src/reg_kernels.h index 31cd49b3da..014c5b7190 100644 --- a/mindspore/lite/src/reg_kernels.h +++ b/mindspore/lite/src/reg_kernels.h @@ -20,13 +20,688 @@ namespace mindspore { namespace kernel { #ifdef MS_COMPILE_IOS -// populate +// base +extern void _kCPUkNumberTypeInt32PrimitiveType_StridedSlice(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_StridedSlice(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_StridedSlice(); +extern void _kCPUkNumberTypeInt8PrimitiveType_StridedSlice(); +extern void _kCPUkNumberTypeUInt8PrimitiveType_QuantDTypeCast(); +extern void _kCPUkNumberTypeInt8PrimitiveType_QuantDTypeCast(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_QuantDTypeCast(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Assert(); +extern void _kCPUkNumberTypeBoolPrimitiveType_Assert(); +extern void _kCPUkNumberTypeInt32PrimitiveType_TensorListReserve(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_TensorListReserve(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_TensorListReserve(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_TensorListStack(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_TensorListStack(); +extern void _kCPUkNumberTypeInt32PrimitiveType_TensorListStack(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_ArgMaxFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_ArgMinFusion(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_ArgMaxFusion(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_ArgMinFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_TensorListGetItem(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_TensorListGetItem(); +extern void _kCPUkNumberTypeInt32PrimitiveType_TensorListGetItem(); +extern void _kCPUkNumberTypeInt32PrimitiveType_RandomStandardNormal(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Split(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Split(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Split(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Switch(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Switch(); +extern void _kCPUkNumberTypeBoolPrimitiveType_Switch(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Switch(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_PriorBox(); +extern void _kCPUkNumberTypeInt8PrimitiveType_PriorBox(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_ConstantOfShape(); +extern void _kCPUkNumberTypeInt32PrimitiveType_ConstantOfShape(); +extern void _kCPUkNumberTypeInt64PrimitiveType_ConstantOfShape(); +extern void _kCPUkNumberTypeInt32PrimitiveType_SliceFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_SliceFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_TensorListSetItem(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_TensorListSetItem(); +extern void _kCPUkNumberTypeInt32PrimitiveType_TensorListSetItem(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Reshape(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Reshape(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Reshape(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Flatten(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Flatten(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_FlattenGrad(); +extern void _kCPUkNumberTypeInt32PrimitiveType_ExpandDims(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_ExpandDims(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_ExpandDims(); +extern void _kCPUkNumberTypeInt8PrimitiveType_ExpandDims(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Squeeze(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Squeeze(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Squeeze(); +extern void _kCPUkNumberTypeBoolPrimitiveType_Squeeze(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Unsqueeze(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Unsqueeze(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Unsqueeze(); +extern void _kCPUkNumberTypeInt64PrimitiveType_Unsqueeze(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_TileFusion(); +extern void _kCPUkNumberTypeInt32PrimitiveType_TileFusion(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_TileFusion(); +extern void _kCPUkNumberTypeBoolPrimitiveType_Select(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_TensorListFromTensor(); +extern void _kCPUkNumberTypeInt32PrimitiveType_TensorListFromTensor(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_TensorListFromTensor(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Stack(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Stack(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Merge(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Merge(); +extern void _kCPUkNumberTypeBoolPrimitiveType_Merge(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Merge(); + +// fp16 +#ifdef ENABLE_FP16 +extern void _kCPUkNumberTypeFloat16PrimitiveType_SliceFusion(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Concat(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_BiasAdd(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Crop(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_ScaleFusion(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_PowFusion(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Stack(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_GRU(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Softmax(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_MatMul(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Transpose(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_NotEqual(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Equal(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Less(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_LessEqual(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Greater(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_GreaterEqual(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_ReduceFusion(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_FullConnection(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Gather(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_LSTM(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Activation(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Cast(); +extern void _kCPUkNumberTypeInt64PrimitiveType_Cast(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Abs(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Cos(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Log(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Square(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Sqrt(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Rsqrt(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Sin(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_LogicalNot(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Floor(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Ceil(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Round(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Neg(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Reciprocal(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Erf(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_AvgPoolFusion(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_MaxPoolFusion(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_QuantDTypeCast(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_PadFusion(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_BatchNorm(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Conv2dTransposeFusion(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_MulFusion(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_AddFusion(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_SubFusion(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_DivFusion(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_FloorMod(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_FloorDiv(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_LogicalAnd(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_LogicalOr(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Maximum(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Minimum(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Eltwise(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_SquaredDifference(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Conv2DFusion(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_InstanceNorm(); +#endif + +// fp32 +extern void _kCPUkNumberTypeFloat32PrimitiveType_ROIPooling(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_PadFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Activation(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_ReverseSequence(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_AdderFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_DetectionPostProcess(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_FullConnection(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_GatherNd(); +extern void _kCPUkNumberTypeInt32PrimitiveType_GatherNd(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Rank(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Elu(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_NonMaxSuppression(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_BatchNorm(); extern void _kCPUkNumberTypeFloat32PrimitiveType_ReduceFusion(); +extern void _kCPUkNumberTypeIntPrimitiveType_ReduceFusion(); +extern void _kCPUkNumberTypeInt32PrimitiveType_ReduceFusion(); extern void _kCPUkNumberTypeBoolPrimitiveType_ReduceFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_SparseToDense(); +extern void _kCPUkNumberTypeInt32PrimitiveType_SparseToDense(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Gather(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Gather(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_DepthToSpace(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Unique(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_ExpFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_ScatterNd(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_SpaceToDepth(); +extern void _kCPUkNumberTypeBoolPrimitiveType_NonZero(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Resize(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_PReLUFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_MulFusion(); +extern void _kCPUkNumberTypeInt32PrimitiveType_MulFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_AddFusion(); +extern void _kCPUkNumberTypeInt32PrimitiveType_AddFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_SubFusion(); +extern void _kCPUkNumberTypeInt32PrimitiveType_SubFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_DivFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_RealDiv(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Mod(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Mod(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_LogicalAnd(); +extern void _kCPUkNumberTypeBoolPrimitiveType_LogicalAnd(); +extern void _kCPUkNumberTypeInt32PrimitiveType_LogicalAnd(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_LogicalOr(); +extern void _kCPUkNumberTypeBoolPrimitiveType_LogicalOr(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Maximum(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Minimum(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Maximum(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Minimum(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_FloorDiv(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_FloorMod(); +extern void _kCPUkNumberTypeInt32PrimitiveType_FloorDiv(); +extern void _kCPUkNumberTypeInt32PrimitiveType_FloorMod(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_SquaredDifference(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Eltwise(); +extern void _kCPUkNumberTypeInt32PrimitiveType_DivFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_AvgPoolFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_MaxPoolFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_EmbeddingLookupFusion(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Size(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Size(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Transpose(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Transpose(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_PowFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Abs(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Cos(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Log(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Square(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Sqrt(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Rsqrt(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Sin(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_LogicalNot(); +extern void _kCPUkNumberTypeBoolPrimitiveType_LogicalNot(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Floor(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Ceil(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Round(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Neg(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Reciprocal(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Erf(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_ReverseV2(); +extern void _kCPUkNumberTypeInt32PrimitiveType_ReverseV2(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Unstack(); +extern void _kCPUkNumberTypeInt32PrimitiveType_InvertPermutation(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_InvertPermutation(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_SpaceToBatch(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_SpaceToBatchND(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_AddN(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_MatMul(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_FusedBatchNorm(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_BiasAdd(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_LRN(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_TopKFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_GRU(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Shape(); +extern void _kCPUkNumberTypeFloat16PrimitiveType_Shape(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Shape(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Shape(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_L2NormalizeFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_ScaleFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_BatchToSpace(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_BatchToSpaceND(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Where(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Where(); +extern void _kCPUkNumberTypeBoolPrimitiveType_Where(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_SkipGram(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_BroadcastTo(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_CropAndResize(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Conv2dTransposeFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Conv2DFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_LayerNormFusion(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_ZerosLike(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_InstanceNorm(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Cast(); +extern void _kCPUkNumberTypeUInt8PrimitiveType_Cast(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Cast(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Cast(); +extern void _kCPUkNumberTypeInt64PrimitiveType_Cast(); +extern void _kCPUkNumberTypeBoolPrimitiveType_Cast(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_LshProjection(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Fill(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Fill(); +extern void _kCPUkNumberTypeInt32PrimitiveType_OneHot(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_LSTM(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Crop(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Crop(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Equal(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Equal(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_NotEqual(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Less(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Less(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_LessEqual(); +extern void _kCPUkNumberTypeInt32PrimitiveType_LessEqual(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Greater(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_GreaterEqual(); +extern void _kCPUkNumberTypeInt32PrimitiveType_GreaterEqual(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Range(); +extern void _kCPUkNumberTypeFloatPrimitiveType_Range(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Range(); +extern void _kCPUkNumberTypeIntPrimitiveType_Range(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Softmax(); +extern void _kCPUkNumberTypeInt32PrimitiveType_Concat(); +extern void _kCPUkNumberTypeFloat32PrimitiveType_Concat(); + +// int8 +extern void _kCPUkNumberTypeInt8PrimitiveType_PowFusion(); +extern void _kCPUkNumberTypeInt8PrimitiveType_ReduceFusion(); +extern void _kCPUkNumberTypeInt8PrimitiveType_TopKFusion(); +extern void _kCPUkNumberTypeInt8PrimitiveType_GatherNd(); +extern void _kCPUkNumberTypeInt8PrimitiveType_PadFusion(); +extern void _kCPUkNumberTypeInt8PrimitiveType_L2NormalizeFusion(); +extern void _kCPUkNumberTypeInt8PrimitiveType_MulFusion(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Unsqueeze(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Conv2DFusion(); +extern void _kCPUkNumberTypeInt8PrimitiveType_AvgPoolFusion(); +extern void _kCPUkNumberTypeInt8PrimitiveType_MaxPoolFusion(); +extern void _kCPUkNumberTypeInt8PrimitiveType_BatchNorm(); +extern void _kCPUkNumberTypeInt8PrimitiveType_FusedBatchNorm(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Equal(); +extern void _kCPUkNumberTypeInt8PrimitiveType_NotEqual(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Less(); +extern void _kCPUkNumberTypeInt8PrimitiveType_LessEqual(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Greater(); +extern void _kCPUkNumberTypeInt8PrimitiveType_GreaterEqual(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Eltwise(); +extern void _kCPUkNumberTypeInt8PrimitiveType_BatchToSpace(); +extern void _kCPUkNumberTypeInt8PrimitiveType_BatchToSpaceND(); +extern void _kCPUkNumberTypeInt8PrimitiveType_DepthToSpace(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Activation(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Round(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Floor(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Ceil(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Abs(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Sin(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Cos(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Log(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Sqrt(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Rsqrt(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Square(); +extern void _kCPUkNumberTypeInt8PrimitiveType_LogicalNot(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Reciprocal(); +extern void _kCPUkNumberTypeInt8PrimitiveType_SliceFusion(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Reshape(); +extern void _kCPUkNumberTypeInt8PrimitiveType_DivFusion(); +extern void _kCPUkNumberTypeInt8PrimitiveType_MatMul(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Concat(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Crop(); +extern void _kCPUkNumberTypeInt8PrimitiveType_SpaceToBatch(); +extern void _kCPUkNumberTypeInt8PrimitiveType_SpaceToBatchND(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Split(); +extern void _kCPUkNumberTypeInt8PrimitiveType_BiasAdd(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Resize(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Gather(); +extern void _kCPUkNumberTypeInt8PrimitiveType_AddFusion(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Squeeze(); +extern void _kCPUkNumberTypeInt8PrimitiveType_ScaleFusion(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Softmax(); +extern void _kCPUkNumberTypeInt8PrimitiveType_ArgMaxFusion(); +extern void _kCPUkNumberTypeInt8PrimitiveType_ArgMinFusion(); +extern void _kCPUkNumberTypeInt8PrimitiveType_FullConnection(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Transpose(); +extern void _kCPUkNumberTypeInt8PrimitiveType_Conv2dTransposeFusion(); +extern void _kCPUkNumberTypeInt8PrimitiveType_DetectionPostProcess(); +extern void _kCPUkNumberTypeInt8PrimitiveType_LeakyRelu(); +extern void _kCPUkNumberTypeInt8PrimitiveType_LayerNormFusion(); +extern void _kCPUkNumberTypeInt8PrimitiveType_SubFusion(); void RegisterKernels() { + // base + _kCPUkNumberTypeInt32PrimitiveType_StridedSlice(); + _kCPUkNumberTypeFloat32PrimitiveType_StridedSlice(); + _kCPUkNumberTypeFloat16PrimitiveType_StridedSlice(); + _kCPUkNumberTypeInt8PrimitiveType_StridedSlice(); + _kCPUkNumberTypeUInt8PrimitiveType_QuantDTypeCast(); + _kCPUkNumberTypeInt8PrimitiveType_QuantDTypeCast(); + _kCPUkNumberTypeFloat32PrimitiveType_QuantDTypeCast(); + _kCPUkNumberTypeFloat32PrimitiveType_Assert(); + _kCPUkNumberTypeBoolPrimitiveType_Assert(); + _kCPUkNumberTypeInt32PrimitiveType_TensorListReserve(); + _kCPUkNumberTypeFloat16PrimitiveType_TensorListReserve(); + _kCPUkNumberTypeFloat32PrimitiveType_TensorListReserve(); + _kCPUkNumberTypeFloat32PrimitiveType_TensorListStack(); + _kCPUkNumberTypeFloat16PrimitiveType_TensorListStack(); + _kCPUkNumberTypeInt32PrimitiveType_TensorListStack(); + _kCPUkNumberTypeFloat32PrimitiveType_ArgMaxFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_ArgMinFusion(); + _kCPUkNumberTypeFloat16PrimitiveType_ArgMaxFusion(); + _kCPUkNumberTypeFloat16PrimitiveType_ArgMinFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_TensorListGetItem(); + _kCPUkNumberTypeFloat16PrimitiveType_TensorListGetItem(); + _kCPUkNumberTypeInt32PrimitiveType_TensorListGetItem(); + _kCPUkNumberTypeInt32PrimitiveType_RandomStandardNormal(); + _kCPUkNumberTypeFloat16PrimitiveType_Split(); + _kCPUkNumberTypeInt32PrimitiveType_Split(); + _kCPUkNumberTypeFloat32PrimitiveType_Split(); + _kCPUkNumberTypeFloat32PrimitiveType_Switch(); + _kCPUkNumberTypeFloat16PrimitiveType_Switch(); + _kCPUkNumberTypeBoolPrimitiveType_Switch(); + _kCPUkNumberTypeInt32PrimitiveType_Switch(); + _kCPUkNumberTypeFloat32PrimitiveType_PriorBox(); + _kCPUkNumberTypeInt8PrimitiveType_PriorBox(); + _kCPUkNumberTypeFloat32PrimitiveType_ConstantOfShape(); + _kCPUkNumberTypeInt32PrimitiveType_ConstantOfShape(); + _kCPUkNumberTypeInt64PrimitiveType_ConstantOfShape(); + _kCPUkNumberTypeInt32PrimitiveType_SliceFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_SliceFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_TensorListSetItem(); + _kCPUkNumberTypeFloat16PrimitiveType_TensorListSetItem(); + _kCPUkNumberTypeInt32PrimitiveType_TensorListSetItem(); + _kCPUkNumberTypeInt32PrimitiveType_Reshape(); + _kCPUkNumberTypeFloat32PrimitiveType_Reshape(); + _kCPUkNumberTypeFloat16PrimitiveType_Reshape(); + _kCPUkNumberTypeFloat16PrimitiveType_Flatten(); + _kCPUkNumberTypeFloat32PrimitiveType_Flatten(); + _kCPUkNumberTypeFloat32PrimitiveType_FlattenGrad(); + _kCPUkNumberTypeInt32PrimitiveType_ExpandDims(); + _kCPUkNumberTypeFloat16PrimitiveType_ExpandDims(); + _kCPUkNumberTypeFloat32PrimitiveType_ExpandDims(); + _kCPUkNumberTypeInt8PrimitiveType_ExpandDims(); + _kCPUkNumberTypeFloat32PrimitiveType_Squeeze(); + _kCPUkNumberTypeFloat16PrimitiveType_Squeeze(); + _kCPUkNumberTypeInt32PrimitiveType_Squeeze(); + _kCPUkNumberTypeBoolPrimitiveType_Squeeze(); + _kCPUkNumberTypeFloat16PrimitiveType_Unsqueeze(); + _kCPUkNumberTypeFloat32PrimitiveType_Unsqueeze(); + _kCPUkNumberTypeInt32PrimitiveType_Unsqueeze(); + _kCPUkNumberTypeInt64PrimitiveType_Unsqueeze(); + _kCPUkNumberTypeFloat32PrimitiveType_TileFusion(); + _kCPUkNumberTypeInt32PrimitiveType_TileFusion(); + _kCPUkNumberTypeFloat16PrimitiveType_TileFusion(); + _kCPUkNumberTypeBoolPrimitiveType_Select(); + _kCPUkNumberTypeFloat32PrimitiveType_TensorListFromTensor(); + _kCPUkNumberTypeInt32PrimitiveType_TensorListFromTensor(); + _kCPUkNumberTypeFloat16PrimitiveType_TensorListFromTensor(); + _kCPUkNumberTypeFloat32PrimitiveType_Stack(); + _kCPUkNumberTypeInt32PrimitiveType_Stack(); + _kCPUkNumberTypeFloat32PrimitiveType_Merge(); + _kCPUkNumberTypeFloat16PrimitiveType_Merge(); + _kCPUkNumberTypeBoolPrimitiveType_Merge(); + _kCPUkNumberTypeInt32PrimitiveType_Merge(); + +// fp16 +#ifdef ENABLE_FP16 + _kCPUkNumberTypeFloat16PrimitiveType_SliceFusion(); + _kCPUkNumberTypeFloat16PrimitiveType_Concat(); + _kCPUkNumberTypeFloat16PrimitiveType_BiasAdd(); + _kCPUkNumberTypeFloat16PrimitiveType_Crop(); + _kCPUkNumberTypeFloat16PrimitiveType_ScaleFusion(); + _kCPUkNumberTypeFloat16PrimitiveType_PowFusion(); + _kCPUkNumberTypeFloat16PrimitiveType_Stack(); + _kCPUkNumberTypeFloat16PrimitiveType_GRU(); + _kCPUkNumberTypeFloat16PrimitiveType_Softmax(); + _kCPUkNumberTypeFloat16PrimitiveType_MatMul(); + _kCPUkNumberTypeFloat16PrimitiveType_Transpose(); + _kCPUkNumberTypeFloat16PrimitiveType_NotEqual(); + _kCPUkNumberTypeFloat16PrimitiveType_Equal(); + _kCPUkNumberTypeFloat16PrimitiveType_Less(); + _kCPUkNumberTypeFloat16PrimitiveType_LessEqual(); + _kCPUkNumberTypeFloat16PrimitiveType_Greater(); + _kCPUkNumberTypeFloat16PrimitiveType_GreaterEqual(); + _kCPUkNumberTypeFloat16PrimitiveType_ReduceFusion(); + _kCPUkNumberTypeFloat16PrimitiveType_FullConnection(); + _kCPUkNumberTypeFloat16PrimitiveType_Gather(); + _kCPUkNumberTypeFloat16PrimitiveType_LSTM(); + _kCPUkNumberTypeFloat16PrimitiveType_Activation(); + _kCPUkNumberTypeFloat16PrimitiveType_Cast(); + _kCPUkNumberTypeInt64PrimitiveType_Cast(); + _kCPUkNumberTypeFloat16PrimitiveType_Abs(); + _kCPUkNumberTypeFloat16PrimitiveType_Cos(); + _kCPUkNumberTypeFloat16PrimitiveType_Log(); + _kCPUkNumberTypeFloat16PrimitiveType_Square(); + _kCPUkNumberTypeFloat16PrimitiveType_Sqrt(); + _kCPUkNumberTypeFloat16PrimitiveType_Rsqrt(); + _kCPUkNumberTypeFloat16PrimitiveType_Sin(); + _kCPUkNumberTypeFloat16PrimitiveType_LogicalNot(); + _kCPUkNumberTypeFloat16PrimitiveType_Floor(); + _kCPUkNumberTypeFloat16PrimitiveType_Ceil(); + _kCPUkNumberTypeFloat16PrimitiveType_Round(); + _kCPUkNumberTypeFloat16PrimitiveType_Neg(); + _kCPUkNumberTypeFloat16PrimitiveType_Reciprocal(); + _kCPUkNumberTypeFloat16PrimitiveType_Erf(); + _kCPUkNumberTypeFloat16PrimitiveType_AvgPoolFusion(); + _kCPUkNumberTypeFloat16PrimitiveType_MaxPoolFusion(); + _kCPUkNumberTypeFloat16PrimitiveType_QuantDTypeCast(); + _kCPUkNumberTypeFloat16PrimitiveType_PadFusion(); + _kCPUkNumberTypeFloat16PrimitiveType_BatchNorm(); + _kCPUkNumberTypeFloat16PrimitiveType_Conv2dTransposeFusion(); + _kCPUkNumberTypeFloat16PrimitiveType_MulFusion(); + _kCPUkNumberTypeFloat16PrimitiveType_AddFusion(); + _kCPUkNumberTypeFloat16PrimitiveType_SubFusion(); + _kCPUkNumberTypeFloat16PrimitiveType_DivFusion(); + _kCPUkNumberTypeFloat16PrimitiveType_FloorMod(); + _kCPUkNumberTypeFloat16PrimitiveType_FloorDiv(); + _kCPUkNumberTypeFloat16PrimitiveType_LogicalAnd(); + _kCPUkNumberTypeFloat16PrimitiveType_LogicalOr(); + _kCPUkNumberTypeFloat16PrimitiveType_Maximum(); + _kCPUkNumberTypeFloat16PrimitiveType_Minimum(); + _kCPUkNumberTypeFloat16PrimitiveType_Eltwise(); + _kCPUkNumberTypeFloat16PrimitiveType_SquaredDifference(); + _kCPUkNumberTypeFloat16PrimitiveType_Conv2DFusion(); + _kCPUkNumberTypeFloat16PrimitiveType_InstanceNorm(); +#endif + + // fp32 + _kCPUkNumberTypeFloat32PrimitiveType_ROIPooling(); + _kCPUkNumberTypeFloat32PrimitiveType_PadFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_Activation(); + _kCPUkNumberTypeFloat32PrimitiveType_ReverseSequence(); + _kCPUkNumberTypeFloat32PrimitiveType_AdderFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_DetectionPostProcess(); + _kCPUkNumberTypeFloat32PrimitiveType_FullConnection(); + _kCPUkNumberTypeFloat32PrimitiveType_GatherNd(); + _kCPUkNumberTypeInt32PrimitiveType_GatherNd(); + _kCPUkNumberTypeFloat32PrimitiveType_Rank(); + _kCPUkNumberTypeFloat32PrimitiveType_Elu(); + _kCPUkNumberTypeFloat32PrimitiveType_NonMaxSuppression(); + _kCPUkNumberTypeFloat32PrimitiveType_BatchNorm(); _kCPUkNumberTypeFloat32PrimitiveType_ReduceFusion(); + _kCPUkNumberTypeIntPrimitiveType_ReduceFusion(); + _kCPUkNumberTypeInt32PrimitiveType_ReduceFusion(); _kCPUkNumberTypeBoolPrimitiveType_ReduceFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_SparseToDense(); + _kCPUkNumberTypeInt32PrimitiveType_SparseToDense(); + _kCPUkNumberTypeFloat32PrimitiveType_Gather(); + _kCPUkNumberTypeInt32PrimitiveType_Gather(); + _kCPUkNumberTypeFloat32PrimitiveType_DepthToSpace(); + _kCPUkNumberTypeFloat32PrimitiveType_Unique(); + _kCPUkNumberTypeFloat32PrimitiveType_ExpFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_ScatterNd(); + _kCPUkNumberTypeFloat32PrimitiveType_SpaceToDepth(); + _kCPUkNumberTypeBoolPrimitiveType_NonZero(); + _kCPUkNumberTypeFloat32PrimitiveType_Resize(); + _kCPUkNumberTypeFloat32PrimitiveType_PReLUFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_MulFusion(); + _kCPUkNumberTypeInt32PrimitiveType_MulFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_AddFusion(); + _kCPUkNumberTypeInt32PrimitiveType_AddFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_SubFusion(); + _kCPUkNumberTypeInt32PrimitiveType_SubFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_DivFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_RealDiv(); + _kCPUkNumberTypeFloat32PrimitiveType_Mod(); + _kCPUkNumberTypeInt32PrimitiveType_Mod(); + _kCPUkNumberTypeFloat32PrimitiveType_LogicalAnd(); + _kCPUkNumberTypeBoolPrimitiveType_LogicalAnd(); + _kCPUkNumberTypeInt32PrimitiveType_LogicalAnd(); + _kCPUkNumberTypeFloat32PrimitiveType_LogicalOr(); + _kCPUkNumberTypeBoolPrimitiveType_LogicalOr(); + _kCPUkNumberTypeFloat32PrimitiveType_Maximum(); + _kCPUkNumberTypeFloat32PrimitiveType_Minimum(); + _kCPUkNumberTypeInt32PrimitiveType_Maximum(); + _kCPUkNumberTypeInt32PrimitiveType_Minimum(); + _kCPUkNumberTypeFloat32PrimitiveType_FloorDiv(); + _kCPUkNumberTypeFloat32PrimitiveType_FloorMod(); + _kCPUkNumberTypeInt32PrimitiveType_FloorDiv(); + _kCPUkNumberTypeInt32PrimitiveType_FloorMod(); + _kCPUkNumberTypeFloat32PrimitiveType_SquaredDifference(); + _kCPUkNumberTypeFloat32PrimitiveType_Eltwise(); + _kCPUkNumberTypeInt32PrimitiveType_DivFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_AvgPoolFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_MaxPoolFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_EmbeddingLookupFusion(); + _kCPUkNumberTypeInt32PrimitiveType_Size(); + _kCPUkNumberTypeFloat32PrimitiveType_Size(); + _kCPUkNumberTypeFloat32PrimitiveType_Transpose(); + _kCPUkNumberTypeInt32PrimitiveType_Transpose(); + _kCPUkNumberTypeFloat32PrimitiveType_PowFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_Abs(); + _kCPUkNumberTypeFloat32PrimitiveType_Cos(); + _kCPUkNumberTypeFloat32PrimitiveType_Log(); + _kCPUkNumberTypeFloat32PrimitiveType_Square(); + _kCPUkNumberTypeFloat32PrimitiveType_Sqrt(); + _kCPUkNumberTypeFloat32PrimitiveType_Rsqrt(); + _kCPUkNumberTypeFloat32PrimitiveType_Sin(); + _kCPUkNumberTypeFloat32PrimitiveType_LogicalNot(); + _kCPUkNumberTypeBoolPrimitiveType_LogicalNot(); + _kCPUkNumberTypeFloat32PrimitiveType_Floor(); + _kCPUkNumberTypeFloat32PrimitiveType_Ceil(); + _kCPUkNumberTypeFloat32PrimitiveType_Round(); + _kCPUkNumberTypeFloat32PrimitiveType_Neg(); + _kCPUkNumberTypeFloat32PrimitiveType_Reciprocal(); + _kCPUkNumberTypeFloat32PrimitiveType_Erf(); + _kCPUkNumberTypeFloat32PrimitiveType_ReverseV2(); + _kCPUkNumberTypeInt32PrimitiveType_ReverseV2(); + _kCPUkNumberTypeFloat32PrimitiveType_Unstack(); + _kCPUkNumberTypeInt32PrimitiveType_InvertPermutation(); + _kCPUkNumberTypeFloat32PrimitiveType_InvertPermutation(); + _kCPUkNumberTypeFloat32PrimitiveType_SpaceToBatch(); + _kCPUkNumberTypeFloat32PrimitiveType_SpaceToBatchND(); + _kCPUkNumberTypeFloat32PrimitiveType_AddN(); + _kCPUkNumberTypeFloat32PrimitiveType_MatMul(); + _kCPUkNumberTypeFloat32PrimitiveType_FusedBatchNorm(); + _kCPUkNumberTypeFloat32PrimitiveType_BiasAdd(); + _kCPUkNumberTypeFloat32PrimitiveType_LRN(); + _kCPUkNumberTypeFloat32PrimitiveType_TopKFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_GRU(); + _kCPUkNumberTypeInt32PrimitiveType_Shape(); + _kCPUkNumberTypeFloat16PrimitiveType_Shape(); + _kCPUkNumberTypeFloat32PrimitiveType_Shape(); + _kCPUkNumberTypeInt8PrimitiveType_Shape(); + _kCPUkNumberTypeFloat32PrimitiveType_L2NormalizeFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_ScaleFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_BatchToSpace(); + _kCPUkNumberTypeFloat32PrimitiveType_BatchToSpaceND(); + _kCPUkNumberTypeInt32PrimitiveType_Where(); + _kCPUkNumberTypeFloat32PrimitiveType_Where(); + _kCPUkNumberTypeBoolPrimitiveType_Where(); + _kCPUkNumberTypeFloat32PrimitiveType_SkipGram(); + _kCPUkNumberTypeFloat32PrimitiveType_BroadcastTo(); + _kCPUkNumberTypeFloat32PrimitiveType_CropAndResize(); + _kCPUkNumberTypeFloat32PrimitiveType_Conv2dTransposeFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_Conv2DFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_LayerNormFusion(); + _kCPUkNumberTypeFloat32PrimitiveType_ZerosLike(); + _kCPUkNumberTypeFloat32PrimitiveType_InstanceNorm(); + _kCPUkNumberTypeFloat32PrimitiveType_Cast(); + _kCPUkNumberTypeUInt8PrimitiveType_Cast(); + _kCPUkNumberTypeInt8PrimitiveType_Cast(); + _kCPUkNumberTypeInt32PrimitiveType_Cast(); + _kCPUkNumberTypeInt64PrimitiveType_Cast(); + _kCPUkNumberTypeBoolPrimitiveType_Cast(); + _kCPUkNumberTypeFloat32PrimitiveType_LshProjection(); + _kCPUkNumberTypeInt32PrimitiveType_Fill(); + _kCPUkNumberTypeFloat32PrimitiveType_Fill(); + _kCPUkNumberTypeInt32PrimitiveType_OneHot(); + _kCPUkNumberTypeFloat32PrimitiveType_LSTM(); + _kCPUkNumberTypeInt32PrimitiveType_Crop(); + _kCPUkNumberTypeFloat32PrimitiveType_Crop(); + _kCPUkNumberTypeFloat32PrimitiveType_Equal(); + _kCPUkNumberTypeInt32PrimitiveType_Equal(); + _kCPUkNumberTypeFloat32PrimitiveType_NotEqual(); + _kCPUkNumberTypeFloat32PrimitiveType_Less(); + _kCPUkNumberTypeInt32PrimitiveType_Less(); + _kCPUkNumberTypeFloat32PrimitiveType_LessEqual(); + _kCPUkNumberTypeInt32PrimitiveType_LessEqual(); + _kCPUkNumberTypeFloat32PrimitiveType_Greater(); + _kCPUkNumberTypeFloat32PrimitiveType_GreaterEqual(); + _kCPUkNumberTypeInt32PrimitiveType_GreaterEqual(); + _kCPUkNumberTypeFloat32PrimitiveType_Range(); + _kCPUkNumberTypeFloatPrimitiveType_Range(); + _kCPUkNumberTypeInt32PrimitiveType_Range(); + _kCPUkNumberTypeIntPrimitiveType_Range(); + _kCPUkNumberTypeFloat32PrimitiveType_Softmax(); + _kCPUkNumberTypeInt32PrimitiveType_Concat(); + _kCPUkNumberTypeFloat32PrimitiveType_Concat(); + + // int8 + _kCPUkNumberTypeInt8PrimitiveType_PowFusion(); + _kCPUkNumberTypeInt8PrimitiveType_ReduceFusion(); + _kCPUkNumberTypeInt8PrimitiveType_TopKFusion(); + _kCPUkNumberTypeInt8PrimitiveType_GatherNd(); + _kCPUkNumberTypeInt8PrimitiveType_PadFusion(); + _kCPUkNumberTypeInt8PrimitiveType_L2NormalizeFusion(); + _kCPUkNumberTypeInt8PrimitiveType_MulFusion(); + _kCPUkNumberTypeInt8PrimitiveType_Unsqueeze(); + _kCPUkNumberTypeInt8PrimitiveType_Conv2DFusion(); + _kCPUkNumberTypeInt8PrimitiveType_AvgPoolFusion(); + _kCPUkNumberTypeInt8PrimitiveType_MaxPoolFusion(); + _kCPUkNumberTypeInt8PrimitiveType_BatchNorm(); + _kCPUkNumberTypeInt8PrimitiveType_FusedBatchNorm(); + _kCPUkNumberTypeInt8PrimitiveType_Equal(); + _kCPUkNumberTypeInt8PrimitiveType_NotEqual(); + _kCPUkNumberTypeInt8PrimitiveType_Less(); + _kCPUkNumberTypeInt8PrimitiveType_LessEqual(); + _kCPUkNumberTypeInt8PrimitiveType_Greater(); + _kCPUkNumberTypeInt8PrimitiveType_GreaterEqual(); + _kCPUkNumberTypeInt8PrimitiveType_Eltwise(); + _kCPUkNumberTypeInt8PrimitiveType_BatchToSpace(); + _kCPUkNumberTypeInt8PrimitiveType_BatchToSpaceND(); + _kCPUkNumberTypeInt8PrimitiveType_DepthToSpace(); + _kCPUkNumberTypeInt8PrimitiveType_Activation(); + _kCPUkNumberTypeInt8PrimitiveType_Round(); + _kCPUkNumberTypeInt8PrimitiveType_Floor(); + _kCPUkNumberTypeInt8PrimitiveType_Ceil(); + _kCPUkNumberTypeInt8PrimitiveType_Abs(); + _kCPUkNumberTypeInt8PrimitiveType_Sin(); + _kCPUkNumberTypeInt8PrimitiveType_Cos(); + _kCPUkNumberTypeInt8PrimitiveType_Log(); + _kCPUkNumberTypeInt8PrimitiveType_Sqrt(); + _kCPUkNumberTypeInt8PrimitiveType_Rsqrt(); + _kCPUkNumberTypeInt8PrimitiveType_Square(); + _kCPUkNumberTypeInt8PrimitiveType_LogicalNot(); + _kCPUkNumberTypeInt8PrimitiveType_Reciprocal(); + _kCPUkNumberTypeInt8PrimitiveType_SliceFusion(); + _kCPUkNumberTypeInt8PrimitiveType_Reshape(); + _kCPUkNumberTypeInt8PrimitiveType_DivFusion(); + _kCPUkNumberTypeInt8PrimitiveType_MatMul(); + _kCPUkNumberTypeInt8PrimitiveType_Concat(); + _kCPUkNumberTypeInt8PrimitiveType_Crop(); + _kCPUkNumberTypeInt8PrimitiveType_SpaceToBatch(); + _kCPUkNumberTypeInt8PrimitiveType_SpaceToBatchND(); + _kCPUkNumberTypeInt8PrimitiveType_Split(); + _kCPUkNumberTypeInt8PrimitiveType_BiasAdd(); + _kCPUkNumberTypeInt8PrimitiveType_Resize(); + _kCPUkNumberTypeInt8PrimitiveType_Gather(); + _kCPUkNumberTypeInt8PrimitiveType_AddFusion(); + _kCPUkNumberTypeInt8PrimitiveType_Squeeze(); + _kCPUkNumberTypeInt8PrimitiveType_ScaleFusion(); + _kCPUkNumberTypeInt8PrimitiveType_Softmax(); + _kCPUkNumberTypeInt8PrimitiveType_ArgMaxFusion(); + _kCPUkNumberTypeInt8PrimitiveType_ArgMinFusion(); + _kCPUkNumberTypeInt8PrimitiveType_FullConnection(); + _kCPUkNumberTypeInt8PrimitiveType_Transpose(); + _kCPUkNumberTypeInt8PrimitiveType_Conv2dTransposeFusion(); + _kCPUkNumberTypeInt8PrimitiveType_DetectionPostProcess(); + _kCPUkNumberTypeInt8PrimitiveType_LeakyRelu(); + _kCPUkNumberTypeInt8PrimitiveType_LayerNormFusion(); + _kCPUkNumberTypeInt8PrimitiveType_SubFusion(); } #endif diff --git a/mindspore/lite/src/reg_ops.cc b/mindspore/lite/src/reg_ops.cc index 7611949a26..fed72fe730 100644 --- a/mindspore/lite/src/reg_ops.cc +++ b/mindspore/lite/src/reg_ops.cc @@ -18,9 +18,317 @@ namespace mindspore { namespace lite { #ifdef MS_COMPILE_IOS // populate +extern void _PrimitiveType_MulFusionSCHEMA_CUR(); +extern void _PrimitiveType_NonMaxSuppressionSCHEMA_CUR(); +extern void _PrimitiveType_TensorListStackSCHEMA_CUR(); +extern void _PrimitiveType_EltwiseSCHEMA_CUR(); +extern void _PrimitiveType_AssignSCHEMA_CUR(); +extern void _PrimitiveType_CropSCHEMA_CUR(); +extern void _PrimitiveType_BatchNormSCHEMA_CUR(); +extern void _PrimitiveType_NonZeroSCHEMA_CUR(); +extern void _PrimitiveType_AddNSCHEMA_CUR(); +extern void _PrimitiveType_MergeSCHEMA_CUR(); +extern void _PrimitiveType_AbsSCHEMA_CUR(); +extern void _PrimitiveType_CosSCHEMA_CUR(); +extern void _PrimitiveType_SinSCHEMA_CUR(); +extern void _PrimitiveType_LogSCHEMA_CUR(); +extern void _PrimitiveType_NegSCHEMA_CUR(); +extern void _PrimitiveType_NegGradSCHEMA_CUR(); +extern void _PrimitiveType_LogGradSCHEMA_CUR(); +extern void _PrimitiveType_SqrtSCHEMA_CUR(); +extern void _PrimitiveType_SquareSCHEMA_CUR(); +extern void _PrimitiveType_RsqrtSCHEMA_CUR(); +extern void _PrimitiveType_LogicalNotSCHEMA_CUR(); +extern void _PrimitiveType_FloorSCHEMA_CUR(); +extern void _PrimitiveType_CeilSCHEMA_CUR(); +extern void _PrimitiveType_RoundSCHEMA_CUR(); +extern void _PrimitiveType_ReciprocalSCHEMA_CUR(); +extern void _PrimitiveType_AssignAddSCHEMA_CUR(); +extern void _PrimitiveType_UniqueSCHEMA_CUR(); +extern void _PrimitiveType_IsFiniteSCHEMA_CUR(); +extern void _PrimitiveType_TileFusionSCHEMA_CUR(); +extern void _PrimitiveType_BinaryCrossEntropyGradSCHEMA_CUR(); +extern void _PrimitiveType_StridedSliceSCHEMA_CUR(); +extern void _PrimitiveType_L2NormalizeFusionSCHEMA_CUR(); +extern void _PrimitiveType_RangeSCHEMA_CUR(); +extern void _PrimitiveType_SizeSCHEMA_CUR(); +extern void _PrimitiveType_CropAndResizeSCHEMA_CUR(); +extern void _PrimitiveType_QuantDTypeCastSCHEMA_CUR(); +extern void _PrimitiveType_InstanceNormSCHEMA_CUR(); +extern void _PrimitiveType_AddFusionSCHEMA_CUR(); +extern void _PrimitiveType_Conv2dTransposeFusionSCHEMA_CUR(); +extern void _PrimitiveType_PowFusionSCHEMA_CUR(); +extern void _PrimitiveType_ReverseV2SCHEMA_CUR(); +extern void _PrimitiveType_BiasAddSCHEMA_CUR(); +extern void _PrimitiveType_TensorListReserveSCHEMA_CUR(); +extern void _PrimitiveType_SpliceSCHEMA_CUR(); +extern void _PrimitiveType_MatMulSCHEMA_CUR(); +extern void _PrimitiveType_ErfSCHEMA_CUR(); +extern void _PrimitiveType_MinimumGradSCHEMA_CUR(); +extern void _PrimitiveType_MaximumGradSCHEMA_CUR(); +extern void _PrimitiveType_RealDivSCHEMA_CUR(); +extern void _PrimitiveType_LogicalAndSCHEMA_CUR(); +extern void _PrimitiveType_LogicalOrSCHEMA_CUR(); +extern void _PrimitiveType_EqualSCHEMA_CUR(); +extern void _PrimitiveType_NotEqualSCHEMA_CUR(); +extern void _PrimitiveType_LessSCHEMA_CUR(); +extern void _PrimitiveType_LessEqualSCHEMA_CUR(); +extern void _PrimitiveType_GreaterSCHEMA_CUR(); +extern void _PrimitiveType_GreaterEqualSCHEMA_CUR(); +extern void _PrimitiveType_MaximumSCHEMA_CUR(); +extern void _PrimitiveType_MinimumSCHEMA_CUR(); +extern void _PrimitiveType_FloorDivSCHEMA_CUR(); +extern void _PrimitiveType_FloorModSCHEMA_CUR(); +extern void _PrimitiveType_ModSCHEMA_CUR(); +extern void _PrimitiveType_SquaredDifferenceSCHEMA_CUR(); +extern void _PrimitiveType_OneHotSCHEMA_CUR(); +extern void _PrimitiveType_WhileSCHEMA_CUR(); +extern void _PrimitiveType_ScatterNdSCHEMA_CUR(); +extern void _PrimitiveType_DepthToSpaceSCHEMA_CUR(); +extern void _PrimitiveType_GatherSCHEMA_CUR(); +extern void _PrimitiveType_AssertSCHEMA_CUR(); +extern void _PrimitiveType_InvertPermutationSCHEMA_CUR(); +extern void _PrimitiveType_OnesLikeSCHEMA_CUR(); +extern void _PrimitiveType_ConcatSCHEMA_CUR(); +extern void _PrimitiveType_ScaleFusionSCHEMA_CUR(); +extern void _PrimitiveType_HashtableLookupSCHEMA_CUR(); +extern void _PrimitiveType_StackSCHEMA_CUR(); +extern void _PrimitiveType_ShapeSCHEMA_CUR(); +extern void _PrimitiveType_SparseToDenseSCHEMA_CUR(); +extern void _PrimitiveType_TensorListFromTensorSCHEMA_CUR(); +extern void _PrimitiveType_ReverseSequenceSCHEMA_CUR(); +extern void _PrimitiveType_UnsqueezeSCHEMA_CUR(); +extern void _PrimitiveType_FlattenSCHEMA_CUR(); +extern void _PrimitiveType_ArgMinFusionSCHEMA_CUR(); +extern void _PrimitiveType_SpaceToBatchSCHEMA_CUR(); +extern void _PrimitiveType_UnstackSCHEMA_CUR(); +extern void _PrimitiveType_BroadcastToSCHEMA_CUR(); +extern void _PrimitiveType_SelectSCHEMA_CUR(); +extern void _PrimitiveType_LinSpaceSCHEMA_CUR(); +extern void _PrimitiveType_BiasAddGradSCHEMA_CUR(); +extern void _PrimitiveType_LayerNormFusionSCHEMA_CUR(); +extern void _PrimitiveType_EmbeddingLookupFusionSCHEMA_CUR(); +extern void _PrimitiveType_CustomExtractFeaturesSCHEMA_CUR(); +extern void _PrimitiveType_ResizeSCHEMA_CUR(); +extern void _PrimitiveType_AdamSCHEMA_CUR(); +extern void _PrimitiveType_ExpandDimsSCHEMA_CUR(); +extern void _PrimitiveType_IfSCHEMA_CUR(); +extern void _PrimitiveType_SoftmaxSCHEMA_CUR(); +extern void _PrimitiveType_BatchToSpaceSCHEMA_CUR(); +extern void _PrimitiveType_BatchToSpaceNDSCHEMA_CUR(); +extern void _PrimitiveType_UniformRealSCHEMA_CUR(); +extern void _PrimitiveType_DetectionPostProcessSCHEMA_CUR(); +extern void _PrimitiveType_GatherNdSCHEMA_CUR(); +extern void _PrimitiveType_GRUSCHEMA_CUR(); +extern void _PrimitiveType_CustomNormalizeSCHEMA_CUR(); +extern void _PrimitiveType_AdderFusionSCHEMA_CUR(); +extern void _PrimitiveType_RandomStandardNormalSCHEMA_CUR(); +extern void _PrimitiveType_FusedBatchNormSCHEMA_CUR(); +extern void _PrimitiveType_AvgPoolFusionSCHEMA_CUR(); +extern void _PrimitiveType_MaxPoolFusionSCHEMA_CUR(); +extern void _PrimitiveType_ArgMaxFusionSCHEMA_CUR(); +extern void _PrimitiveType_ExpFusionSCHEMA_CUR(); +extern void _PrimitiveType_LayerNormGradSCHEMA_CUR(); +extern void _PrimitiveType_AudioSpectrogramSCHEMA_CUR(); +extern void _PrimitiveType_SpaceToBatchNDSCHEMA_CUR(); +extern void _PrimitiveType_ROIPoolingSCHEMA_CUR(); +extern void _PrimitiveType_PReLUFusionSCHEMA_CUR(); +extern void _PrimitiveType_WhereSCHEMA_CUR(); +extern void _PrimitiveType_SparseSoftmaxCrossEntropyWithLogitsSCHEMA_CUR(); +extern void _PrimitiveType_PartialFusionSCHEMA_CUR(); +extern void _PrimitiveType_SubFusionSCHEMA_CUR(); +extern void _PrimitiveType_DivFusionSCHEMA_CUR(); +extern void _PrimitiveType_MfccSCHEMA_CUR(); +extern void _PrimitiveType_SplitSCHEMA_CUR(); +extern void _PrimitiveType_ActivationSCHEMA_CUR(); +extern void _PrimitiveType_TopKFusionSCHEMA_CUR(); +extern void _PrimitiveType_CastSCHEMA_CUR(); +extern void _PrimitiveType_StridedSliceGradSCHEMA_CUR(); +extern void _PrimitiveType_SliceFusionSCHEMA_CUR(); +extern void _PrimitiveType_TransposeSCHEMA_CUR(); extern void _PrimitiveType_ReduceFusionSCHEMA_CUR(); +extern void _PrimitiveType_FullConnectionSCHEMA_CUR(); +extern void _PrimitiveType_LshProjectionSCHEMA_CUR(); +extern void _PrimitiveType_SkipGramSCHEMA_CUR(); +extern void _PrimitiveType_SqueezeSCHEMA_CUR(); +extern void _PrimitiveType_ActivationGradSCHEMA_CUR(); +extern void _PrimitiveType_SwitchSCHEMA_CUR(); +extern void _PrimitiveType_LSTMSCHEMA_CUR(); +extern void _PrimitiveType_BinaryCrossEntropySCHEMA_CUR(); +extern void _PrimitiveType_EluSCHEMA_CUR(); +extern void _PrimitiveType_Conv2DFusionSCHEMA_CUR(); +extern void _PrimitiveType_TensorListGetItemSCHEMA_CUR(); +extern void _PrimitiveType_LRNSCHEMA_CUR(); +extern void _PrimitiveType_ReshapeSCHEMA_CUR(); +extern void _PrimitiveType_ConstantOfShapeSCHEMA_CUR(); +extern void _PrimitiveType_CustomPredictSCHEMA_CUR(); +extern void _PrimitiveType_PadFusionSCHEMA_CUR(); +extern void _PrimitiveType_PriorBoxSCHEMA_CUR(); +extern void _PrimitiveType_ZerosLikeSCHEMA_CUR(); +extern void _PrimitiveType_DependSCHEMA_CUR(); +extern void _PrimitiveType_UnsortedSegmentSumSCHEMA_CUR(); +extern void _PrimitiveType_TensorListSetItemSCHEMA_CUR(); +extern void _PrimitiveType_FillSCHEMA_CUR(); +extern void _PrimitiveType_RankSCHEMA_CUR(); +extern void _PrimitiveType_ClipSCHEMA_CUR(); +extern void _PrimitiveType_SpaceToDepthSCHEMA_CUR(); -void RegisterOps() { _PrimitiveType_ReduceFusionSCHEMA_CUR(); } +void RegisterOps() { + _PrimitiveType_MulFusionSCHEMA_CUR(); + _PrimitiveType_NonMaxSuppressionSCHEMA_CUR(); + _PrimitiveType_TensorListStackSCHEMA_CUR(); + _PrimitiveType_EltwiseSCHEMA_CUR(); + _PrimitiveType_AssignSCHEMA_CUR(); + _PrimitiveType_CropSCHEMA_CUR(); + _PrimitiveType_BatchNormSCHEMA_CUR(); + _PrimitiveType_NonZeroSCHEMA_CUR(); + _PrimitiveType_AddNSCHEMA_CUR(); + _PrimitiveType_MergeSCHEMA_CUR(); + _PrimitiveType_AbsSCHEMA_CUR(); + _PrimitiveType_CosSCHEMA_CUR(); + _PrimitiveType_SinSCHEMA_CUR(); + _PrimitiveType_LogSCHEMA_CUR(); + _PrimitiveType_NegSCHEMA_CUR(); + _PrimitiveType_NegGradSCHEMA_CUR(); + _PrimitiveType_LogGradSCHEMA_CUR(); + _PrimitiveType_SqrtSCHEMA_CUR(); + _PrimitiveType_SquareSCHEMA_CUR(); + _PrimitiveType_RsqrtSCHEMA_CUR(); + _PrimitiveType_LogicalNotSCHEMA_CUR(); + _PrimitiveType_FloorSCHEMA_CUR(); + _PrimitiveType_CeilSCHEMA_CUR(); + _PrimitiveType_RoundSCHEMA_CUR(); + _PrimitiveType_ReciprocalSCHEMA_CUR(); + _PrimitiveType_AssignAddSCHEMA_CUR(); + _PrimitiveType_UniqueSCHEMA_CUR(); + _PrimitiveType_IsFiniteSCHEMA_CUR(); + _PrimitiveType_TileFusionSCHEMA_CUR(); + _PrimitiveType_BinaryCrossEntropyGradSCHEMA_CUR(); + _PrimitiveType_StridedSliceSCHEMA_CUR(); + _PrimitiveType_L2NormalizeFusionSCHEMA_CUR(); + _PrimitiveType_RangeSCHEMA_CUR(); + _PrimitiveType_SizeSCHEMA_CUR(); + _PrimitiveType_CropAndResizeSCHEMA_CUR(); + _PrimitiveType_QuantDTypeCastSCHEMA_CUR(); + _PrimitiveType_InstanceNormSCHEMA_CUR(); + _PrimitiveType_AddFusionSCHEMA_CUR(); + _PrimitiveType_Conv2dTransposeFusionSCHEMA_CUR(); + _PrimitiveType_PowFusionSCHEMA_CUR(); + _PrimitiveType_ReverseV2SCHEMA_CUR(); + _PrimitiveType_BiasAddSCHEMA_CUR(); + _PrimitiveType_TensorListReserveSCHEMA_CUR(); + _PrimitiveType_SpliceSCHEMA_CUR(); + _PrimitiveType_MatMulSCHEMA_CUR(); + _PrimitiveType_ErfSCHEMA_CUR(); + _PrimitiveType_MinimumGradSCHEMA_CUR(); + _PrimitiveType_MaximumGradSCHEMA_CUR(); + _PrimitiveType_RealDivSCHEMA_CUR(); + _PrimitiveType_LogicalAndSCHEMA_CUR(); + _PrimitiveType_LogicalOrSCHEMA_CUR(); + _PrimitiveType_EqualSCHEMA_CUR(); + _PrimitiveType_NotEqualSCHEMA_CUR(); + _PrimitiveType_LessSCHEMA_CUR(); + _PrimitiveType_LessEqualSCHEMA_CUR(); + _PrimitiveType_GreaterSCHEMA_CUR(); + _PrimitiveType_GreaterEqualSCHEMA_CUR(); + _PrimitiveType_MaximumSCHEMA_CUR(); + _PrimitiveType_MinimumSCHEMA_CUR(); + _PrimitiveType_FloorDivSCHEMA_CUR(); + _PrimitiveType_FloorModSCHEMA_CUR(); + _PrimitiveType_ModSCHEMA_CUR(); + _PrimitiveType_SquaredDifferenceSCHEMA_CUR(); + _PrimitiveType_OneHotSCHEMA_CUR(); + _PrimitiveType_WhileSCHEMA_CUR(); + _PrimitiveType_ScatterNdSCHEMA_CUR(); + _PrimitiveType_DepthToSpaceSCHEMA_CUR(); + _PrimitiveType_GatherSCHEMA_CUR(); + _PrimitiveType_AssertSCHEMA_CUR(); + _PrimitiveType_InvertPermutationSCHEMA_CUR(); + _PrimitiveType_OnesLikeSCHEMA_CUR(); + _PrimitiveType_ConcatSCHEMA_CUR(); + _PrimitiveType_ScaleFusionSCHEMA_CUR(); + _PrimitiveType_HashtableLookupSCHEMA_CUR(); + _PrimitiveType_StackSCHEMA_CUR(); + _PrimitiveType_ShapeSCHEMA_CUR(); + _PrimitiveType_SparseToDenseSCHEMA_CUR(); + _PrimitiveType_TensorListFromTensorSCHEMA_CUR(); + _PrimitiveType_ReverseSequenceSCHEMA_CUR(); + _PrimitiveType_UnsqueezeSCHEMA_CUR(); + _PrimitiveType_FlattenSCHEMA_CUR(); + _PrimitiveType_ArgMinFusionSCHEMA_CUR(); + _PrimitiveType_SpaceToBatchSCHEMA_CUR(); + _PrimitiveType_UnstackSCHEMA_CUR(); + _PrimitiveType_BroadcastToSCHEMA_CUR(); + _PrimitiveType_SelectSCHEMA_CUR(); + _PrimitiveType_LinSpaceSCHEMA_CUR(); + _PrimitiveType_BiasAddGradSCHEMA_CUR(); + _PrimitiveType_LayerNormFusionSCHEMA_CUR(); + _PrimitiveType_EmbeddingLookupFusionSCHEMA_CUR(); + _PrimitiveType_CustomExtractFeaturesSCHEMA_CUR(); + _PrimitiveType_ResizeSCHEMA_CUR(); + _PrimitiveType_AdamSCHEMA_CUR(); + _PrimitiveType_ExpandDimsSCHEMA_CUR(); + _PrimitiveType_IfSCHEMA_CUR(); + _PrimitiveType_SoftmaxSCHEMA_CUR(); + _PrimitiveType_BatchToSpaceSCHEMA_CUR(); + _PrimitiveType_BatchToSpaceNDSCHEMA_CUR(); + _PrimitiveType_UniformRealSCHEMA_CUR(); + _PrimitiveType_DetectionPostProcessSCHEMA_CUR(); + _PrimitiveType_GatherNdSCHEMA_CUR(); + _PrimitiveType_GRUSCHEMA_CUR(); + _PrimitiveType_CustomNormalizeSCHEMA_CUR(); + _PrimitiveType_AdderFusionSCHEMA_CUR(); + _PrimitiveType_RandomStandardNormalSCHEMA_CUR(); + _PrimitiveType_FusedBatchNormSCHEMA_CUR(); + _PrimitiveType_AvgPoolFusionSCHEMA_CUR(); + _PrimitiveType_MaxPoolFusionSCHEMA_CUR(); + _PrimitiveType_ArgMaxFusionSCHEMA_CUR(); + _PrimitiveType_ExpFusionSCHEMA_CUR(); + _PrimitiveType_LayerNormGradSCHEMA_CUR(); + _PrimitiveType_AudioSpectrogramSCHEMA_CUR(); + _PrimitiveType_SpaceToBatchNDSCHEMA_CUR(); + _PrimitiveType_ROIPoolingSCHEMA_CUR(); + _PrimitiveType_PReLUFusionSCHEMA_CUR(); + _PrimitiveType_WhereSCHEMA_CUR(); + _PrimitiveType_SparseSoftmaxCrossEntropyWithLogitsSCHEMA_CUR(); + _PrimitiveType_PartialFusionSCHEMA_CUR(); + _PrimitiveType_SubFusionSCHEMA_CUR(); + _PrimitiveType_DivFusionSCHEMA_CUR(); + _PrimitiveType_MfccSCHEMA_CUR(); + _PrimitiveType_SplitSCHEMA_CUR(); + _PrimitiveType_ActivationSCHEMA_CUR(); + _PrimitiveType_TopKFusionSCHEMA_CUR(); + _PrimitiveType_CastSCHEMA_CUR(); + _PrimitiveType_StridedSliceGradSCHEMA_CUR(); + _PrimitiveType_SliceFusionSCHEMA_CUR(); + _PrimitiveType_TransposeSCHEMA_CUR(); + _PrimitiveType_ReduceFusionSCHEMA_CUR(); + _PrimitiveType_FullConnectionSCHEMA_CUR(); + _PrimitiveType_LshProjectionSCHEMA_CUR(); + _PrimitiveType_SkipGramSCHEMA_CUR(); + _PrimitiveType_SqueezeSCHEMA_CUR(); + _PrimitiveType_ActivationGradSCHEMA_CUR(); + _PrimitiveType_SwitchSCHEMA_CUR(); + _PrimitiveType_LSTMSCHEMA_CUR(); + _PrimitiveType_BinaryCrossEntropySCHEMA_CUR(); + _PrimitiveType_EluSCHEMA_CUR(); + _PrimitiveType_Conv2DFusionSCHEMA_CUR(); + _PrimitiveType_TensorListGetItemSCHEMA_CUR(); + _PrimitiveType_LRNSCHEMA_CUR(); + _PrimitiveType_ReshapeSCHEMA_CUR(); + _PrimitiveType_ConstantOfShapeSCHEMA_CUR(); + _PrimitiveType_CustomPredictSCHEMA_CUR(); + _PrimitiveType_PadFusionSCHEMA_CUR(); + _PrimitiveType_PriorBoxSCHEMA_CUR(); + _PrimitiveType_ZerosLikeSCHEMA_CUR(); + _PrimitiveType_DependSCHEMA_CUR(); + _PrimitiveType_UnsortedSegmentSumSCHEMA_CUR(); + _PrimitiveType_TensorListSetItemSCHEMA_CUR(); + _PrimitiveType_FillSCHEMA_CUR(); + _PrimitiveType_RankSCHEMA_CUR(); + _PrimitiveType_ClipSCHEMA_CUR(); + _PrimitiveType_SpaceToDepthSCHEMA_CUR(); +} #endif } // namespace lite