Huawei_Technology
/
mindspore
Not watched
1
0
Fork 0
Code
Releases 13 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: dbf02f0756
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from 'dbf02f0756'
${ noResults }
mindspore/mindspore/lite/nnacl/infer/infer_register.c

413 lines
16 kB
Raw Blame History 

  1. /**

  2.  * Copyright 2021 Huawei Technologies Co., Ltd

  3.  *

  4.  * Licensed under the Apache License, Version 2.0 (the "License");

  5.  * you may not use this file except in compliance with the License.

  6.  * You may obtain a copy of the License at

  7.  *

  8.  * http://www.apache.org/licenses/LICENSE-2.0

  9.  *

  10.  * Unless required by applicable law or agreed to in writing, software

  11.  * distributed under the License is distributed on an "AS IS" BASIS,

  12.  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  13.  * See the License for the specific language governing permissions and

  14.  * limitations under the License.

  15.  */

  16. #include "nnacl/infer/infer_register.h"

  17. #ifdef MS_COMPILE_IOS

  18. extern void _AudioSpectrogramPrimType_AudioSpectrogram();

  19. extern void _EqualPrimType_Equal();

  20. extern void _GreaterPrimType_Greater();

  21. extern void _GreaterEqualPrimType_GreaterEqual();

  22. extern void _LessPrimType_Less();

  23. extern void _LessEqualPrimType_LessEqual();

  24. extern void _NotEqualPrimType_NotEqual();

  25. extern void _DivGradPrimType_DivGrad();

  26. extern void _MulGradPrimType_MulGrad();

  27. extern void _MinimumGradPrimType_MinimumGrad();

  28. extern void _FullConnectionPrimType_FullConnection();

  29. extern void _AdamPrimType_Adam();

  30. extern void _SlicePrimType_SliceFusion();

  31. extern void _LSTMPrimType_LSTM();

  32. extern void _TilePrimType_TileFusion();

  33. extern void _EmbeddingLookupPrimType_EmbeddingLookupFusion();

  34. extern void _AssignAddPrimType_AssignAdd();

  35. extern void _LinSpacePrimType_LinSpace();

  36. extern void _AddGradPrimType_AddGrad();

  37. extern void _SubGradPrimType_SubGrad();

  38. extern void _OneHotPrimType_OneHot();

  39. extern void _RandomStandardNormalPrimType_RandomStandardNormal();

  40. extern void _StridedSlicePrimType_StridedSlice();

  41. extern void _FillPrimType_Fill();

  42. extern void _CastPrimType_Cast();

  43. extern void _SGDPrimType_SGD();

  44. extern void _UniquePrimType_Unique();

  45. extern void _BatchToSpacePrimType_BatchToSpace();

  46. extern void _TensorListFromTensorPrimType_TensorListFromTensor();

  47. extern void _TensorListGetItemPrimType_TensorListGetItem();

  48. extern void _SoftmaxCrossEntropyWithLogitsPrimType_SoftmaxCrossEntropyWithLogits();

  49. extern void _ExpandDimsPrimType_ExpandDims();

  50. extern void _InvertPermutationPrimType_InvertPermutation();

  51. extern void _MergePrimType_Merge();

  52. extern void _RfftPrimType_Rfft();

  53. extern void _Conv2DBackpropInputFusionPrimType_Conv2DBackpropInputFusion();

  54. extern void _ScatterNdPrimType_ScatterNd();

  55. extern void _FusedBatchNormPrimType_FusedBatchNorm();

  56. extern void _PartialPrimType_PartialFusion();

  57. extern void _HashtableLookupPrimType_HashtableLookup();

  58. extern void _ReducePrimType_ReduceFusion();

  59. extern void _GatherPrimType_Gather();

  60. extern void _SplitPrimType_Split();

  61. extern void _RangePrimType_Range();

  62. extern void _BroadcastToPrimType_BroadcastTo();

  63. extern void _UnsortedSegmentSumPrimType_UnsortedSegmentSum();

  64. extern void _SqueezePrimType_Squeeze();

  65. extern void _MaximumGradPrimType_MaximumGrad();

  66. extern void _PowPrimType_PowFusion();

  67. extern void _PriorBoxPrimType_PriorBox();

  68. extern void _SpaceToDepthPrimType_SpaceToDepth();

  69. extern void _RankPrimType_Rank();

  70. extern void _MfccPrimType_Mfcc();

  71. extern void _BinaryCrossEntropyPrimType_BinaryCrossEntropy();

  72. extern void _Conv2DBackpropFilterFusionPrimType_Conv2DBackpropFilterFusion();

  73. extern void _WherePrimType_Where();

  74. extern void _DetectionPostProcessPrimType_DetectionPostProcess();

  75. extern void _FlattenGradPrimType_FlattenGrad();

  76. extern void _ConstantOfShapePrimType_ConstantOfShape();

  77. extern void _SelectPrimType_Select();

  78. extern void _ConcatPrimType_Concat();

  79. extern void _StackPrimType_Stack();

  80. extern void _SwitchPrimType_Switch();

  81. extern void _NonMaxSuppressionPrimType_NonMaxSuppression();

  82. extern void _SkipGramPrimType_SkipGram();

  83. extern void _AddNPrimType_AddN();

  84. extern void _ArgMinPrimType_ArgMinFusion();

  85. extern void _ArgMaxPrimType_ArgMaxFusion();

  86. extern void _LayerNormFusionPrimType_LayerNormFusion();

  87. extern void _MaxPoolPrimType_MaxPoolFusion();

  88. extern void _AvgPoolPrimType_AvgPoolFusion();

  89. extern void _AssignPrimType_Assign();

  90. extern void _CropPrimType_Crop();

  91. extern void _UnsqueezePrimType_Unsqueeze();

  92. extern void _AbsPrimType_Abs();

  93. extern void _AbsGradPrimType_AbsGrad();

  94. extern void _ActivationPrimType_Activation();

  95. extern void _ActivationGradPrimType_ActivationGrad();

  96. extern void _BatchNormPrimType_BatchNorm();

  97. extern void _BinaryCrossEntropyGradPrimType_BinaryCrossEntropyGrad();

  98. extern void _BiasAddPrimType_BiasAdd();

  99. extern void _CeilPrimType_Ceil();

  100. extern void _ClipPrimType_Clip();

  101. extern void _ControlDependPrimType_ControlDepend();

  102. extern void _CosPrimType_Cos();

  103. extern void _DependPrimType_Depend();

  104. extern void _EluPrimType_Elu();

  105. extern void _ErfPrimType_Erf();

  106. extern void _ExpPrimType_ExpFusion();

  107. extern void _FakeQuantWithMinMaxVarsPrimType_FakeQuantWithMinMaxVars();

  108. extern void _FloorPrimType_Floor();

  109. extern void _IfPrimType_If();

  110. extern void _InstanceNormPrimType_InstanceNorm();

  111. extern void _IsFinitePrimType_IsFinite();

  112. extern void _LeakyReluPrimType_LeakyRelu();

  113. extern void _LogPrimType_Log();

  114. extern void _LogGradPrimType_LogGrad();

  115. extern void _LogicalNotPrimType_LogicalNot();

  116. extern void _LRNPrimType_LRN();

  117. extern void _L2NormalizePrimType_L2NormalizeFusion();

  118. extern void _NegPrimType_Neg();

  119. extern void _NegGradPrimType_NegGrad();

  120. extern void _PowerGradPrimType_PowerGrad();

  121. extern void _PReLUPrimType_PReLUFusion();

  122. extern void _ReciprocalPrimType_Reciprocal();

  123. extern void _ReverseSequencePrimType_ReverseSequence();

  124. extern void _ReversePrimType_ReverseV2();

  125. extern void _RoundPrimType_Round();

  126. extern void _RsqrtPrimType_Rsqrt();

  127. extern void _ScalePrimType_ScaleFusion();

  128. extern void _SigmoidCrossEntropyWithLogitsPrimType_SigmoidCrossEntropyWithLogits();

  129. extern void _SigmoidCrossEntropyWithLogitsGradPrimType_SigmoidCrossEntropyWithLogitsGrad();

  130. extern void _SinPrimType_Sin();

  131. extern void _SmoothL1LossPrimType_SmoothL1Loss();

  132. extern void _SmoothL1LossGradPrimType_SmoothL1LossGrad();

  133. extern void _SqrtPrimType_Sqrt();

  134. extern void _SquarePrimType_Square();

  135. extern void _ZerosLikePrimType_ZerosLike();

  136. extern void _AssertPrimType_Assert();

  137. extern void _TensorListSetItemPrimType_TensorListSetItem();

  138. extern void _SplicePrimType_Splice();

  139. extern void _SparseSoftmaxCrossEntropyWithLogitsPrimType_SparseSoftmaxCrossEntropyWithLogits();

  140. extern void _GRUPrimType_GRU();

  141. extern void _SizeOpPrimType_Size();

  142. extern void _CustomExtractFeaturesPrimType_CustomExtractFeatures();

  143. extern void _CustomPredictPrimType_CustomPredict();

  144. extern void _WhilePrimType_While();

  145. extern void _StridedSliceGradPrimType_StridedSliceGrad();

  146. extern void _LshProjectionPrimType_LshProjection();

  147. extern void _SoftmaxPrimType_Softmax();

  148. extern void _CustomNormalizePrimType_CustomNormalize();

  149. extern void _UnstackPrimType_Unstack();

  150. extern void _ROIPoolingPrimType_ROIPooling();

  151. extern void _LayerNormGradPrimType_LayerNormGrad();

  152. extern void _DropoutGradPrimType_DropoutGrad();

  153. extern void _TopKPrimType_TopKFusion();

  154. extern void _ApplyMomentumPrimType_ApplyMomentum();

  155. extern void _AdderPrimType_AdderFusion();

  156. extern void _Conv2DPrimType_Conv2DFusion();

  157. extern void _UniformRealPrimType_UniformReal();

  158. extern void _AvgPoolGradPrimType_AvgPoolGrad();

  159. extern void _MaxPoolGradPrimType_MaxPoolGrad();

  160. extern void _DepthToSpacePrimType_DepthToSpace();

  161. extern void _Conv2dTransposePrimType_Conv2dTransposeFusion();

  162. extern void _QuantDTypeCastPrimType_QuantDTypeCast();

  163. extern void _FftImagPrimType_FftImag();

  164. extern void _FftRealPrimType_FftReal();

  165. extern void _ResizePrimType_Resize();

  166. extern void _SpaceToBatchNDPrimType_SpaceToBatchND();

  167. extern void _TransposePrimType_Transpose();

  168. extern void _TensorListReservePrimType_TensorListReserve();

  169. extern void _ShapePrimType_Shape();

  170. extern void _BiasAddGradPrimType_BiasAddGrad();

  171. extern void _MatMulPrimType_MatMul();

  172. extern void _DropoutPrimType_Dropout();

  173. extern void _ReshapePrimType_Reshape();

  174. extern void _PadPrimType_PadFusion();

  175. extern void _TensorListStackPrimType_TensorListStack();

  176. extern void _FlattenPrimType_Flatten();

  177. extern void _BatchNormGradPrimType_BatchNormGrad();

  178. extern void _SpaceToBatchPrimType_SpaceToBatch();

  179. extern void _GatherNdPrimType_GatherNd();

  180. extern void _CropAndResizePrimType_CropAndResize();

  181. extern void _SparseToDensePrimType_SparseToDense();

  182. extern void _AddPrimType_AddFusion();

  183. extern void _DivPrimType_DivFusion();

  184. extern void _EltwisePrimType_Eltwise();

  185. extern void _FloorDivPrimType_FloorDiv();

  186. extern void _FloorModPrimType_FloorMod();

  187. extern void _LogicalAndPrimType_LogicalAnd();

  188. extern void _LogicalOrPrimType_LogicalOr();

  189. extern void _MaximumPrimType_Maximum();

  190. extern void _MinimumPrimType_Minimum();

  191. extern void _ModPrimType_Mod();

  192. extern void _MulPrimType_MulFusion();

  193. extern void _RealDivPrimType_RealDiv();

  194. extern void _SubPrimType_SubFusion();

  195. extern void _SquaredDifferencePrimType_SquaredDifference();

  196. 

  197. void RegisterInfer() {

  198.   _AudioSpectrogramPrimType_AudioSpectrogram();

  199.   _EqualPrimType_Equal();

  200.   _GreaterPrimType_Greater();

  201.   _GreaterEqualPrimType_GreaterEqual();

  202.   _LessPrimType_Less();

  203.   _LessEqualPrimType_LessEqual();

  204.   _NotEqualPrimType_NotEqual();

  205.   _DivGradPrimType_DivGrad();

  206.   _MulGradPrimType_MulGrad();

  207.   _MinimumGradPrimType_MinimumGrad();

  208.   _FullConnectionPrimType_FullConnection();

  209.   _AdamPrimType_Adam();

  210.   _SlicePrimType_SliceFusion();

  211.   _LSTMPrimType_LSTM();

  212.   _TilePrimType_TileFusion();

  213.   _EmbeddingLookupPrimType_EmbeddingLookupFusion();

  214.   _AssignAddPrimType_AssignAdd();

  215.   _LinSpacePrimType_LinSpace();

  216.   _AddGradPrimType_AddGrad();

  217.   _SubGradPrimType_SubGrad();

  218.   _OneHotPrimType_OneHot();

  219.   _RandomStandardNormalPrimType_RandomStandardNormal();

  220.   _StridedSlicePrimType_StridedSlice();

  221.   _FillPrimType_Fill();

  222.   _CastPrimType_Cast();

  223.   _SGDPrimType_SGD();

  224.   _UniquePrimType_Unique();

  225.   _BatchToSpacePrimType_BatchToSpace();

  226.   _TensorListFromTensorPrimType_TensorListFromTensor();

  227.   _TensorListGetItemPrimType_TensorListGetItem();

  228.   _SoftmaxCrossEntropyWithLogitsPrimType_SoftmaxCrossEntropyWithLogits();

  229.   _ExpandDimsPrimType_ExpandDims();

  230.   _InvertPermutationPrimType_InvertPermutation();

  231.   _MergePrimType_Merge();

  232.   _RfftPrimType_Rfft();

  233.   _Conv2DBackpropInputFusionPrimType_Conv2DBackpropInputFusion();

  234.   _ScatterNdPrimType_ScatterNd();

  235.   _FusedBatchNormPrimType_FusedBatchNorm();

  236.   _PartialPrimType_PartialFusion();

  237.   _HashtableLookupPrimType_HashtableLookup();

  238.   _ReducePrimType_ReduceFusion();

  239.   _GatherPrimType_Gather();

  240.   _SplitPrimType_Split();

  241.   _RangePrimType_Range();

  242.   _BroadcastToPrimType_BroadcastTo();

  243.   _UnsortedSegmentSumPrimType_UnsortedSegmentSum();

  244.   _SqueezePrimType_Squeeze();

  245.   _MaximumGradPrimType_MaximumGrad();

  246.   _PowPrimType_PowFusion();

  247.   _PriorBoxPrimType_PriorBox();

  248.   _SpaceToDepthPrimType_SpaceToDepth();

  249.   _RankPrimType_Rank();

  250.   _MfccPrimType_Mfcc();

  251.   _BinaryCrossEntropyPrimType_BinaryCrossEntropy();

  252.   _Conv2DBackpropFilterFusionPrimType_Conv2DBackpropFilterFusion();

  253.   _WherePrimType_Where();

  254.   _DetectionPostProcessPrimType_DetectionPostProcess();

  255.   _FlattenGradPrimType_FlattenGrad();

  256.   _ConstantOfShapePrimType_ConstantOfShape();

  257.   _SelectPrimType_Select();

  258.   _ConcatPrimType_Concat();

  259.   _StackPrimType_Stack();

  260.   _SwitchPrimType_Switch();

  261.   _NonMaxSuppressionPrimType_NonMaxSuppression();

  262.   _SkipGramPrimType_SkipGram();

  263.   _AddNPrimType_AddN();

  264.   _ArgMinPrimType_ArgMinFusion();

  265.   _ArgMaxPrimType_ArgMaxFusion();

  266.   _LayerNormFusionPrimType_LayerNormFusion();

  267.   _MaxPoolPrimType_MaxPoolFusion();

  268.   _AvgPoolPrimType_AvgPoolFusion();

  269.   _AssignPrimType_Assign();

  270.   _CropPrimType_Crop();

  271.   _UnsqueezePrimType_Unsqueeze();

  272.   _AbsPrimType_Abs();

  273.   _AbsGradPrimType_AbsGrad();

  274.   _ActivationPrimType_Activation();

  275.   _ActivationGradPrimType_ActivationGrad();

  276.   _BatchNormPrimType_BatchNorm();

  277.   _BinaryCrossEntropyGradPrimType_BinaryCrossEntropyGrad();

  278.   _BiasAddPrimType_BiasAdd();

  279.   _CeilPrimType_Ceil();

  280.   _ClipPrimType_Clip();

  281.   _ControlDependPrimType_ControlDepend();

  282.   _CosPrimType_Cos();

  283.   _DependPrimType_Depend();

  284.   _EluPrimType_Elu();

  285.   _ErfPrimType_Erf();

  286.   _ExpPrimType_ExpFusion();

  287.   _FakeQuantWithMinMaxVarsPrimType_FakeQuantWithMinMaxVars();

  288.   _FloorPrimType_Floor();

  289.   _IfPrimType_If();

  290.   _InstanceNormPrimType_InstanceNorm();

  291.   _IsFinitePrimType_IsFinite();

  292.   _LeakyReluPrimType_LeakyRelu();

  293.   _LogPrimType_Log();

  294.   _LogGradPrimType_LogGrad();

  295.   _LogicalNotPrimType_LogicalNot();

  296.   _LRNPrimType_LRN();

  297.   _L2NormalizePrimType_L2NormalizeFusion();

  298.   _NegPrimType_Neg();

  299.   _NegGradPrimType_NegGrad();

  300.   _PowerGradPrimType_PowerGrad();

  301.   _PReLUPrimType_PReLUFusion();

  302.   _ReciprocalPrimType_Reciprocal();

  303.   _ReverseSequencePrimType_ReverseSequence();

  304.   _ReversePrimType_ReverseV2();

  305.   _RoundPrimType_Round();

  306.   _RsqrtPrimType_Rsqrt();

  307.   _ScalePrimType_ScaleFusion();

  308.   _SigmoidCrossEntropyWithLogitsPrimType_SigmoidCrossEntropyWithLogits();

  309.   _SigmoidCrossEntropyWithLogitsGradPrimType_SigmoidCrossEntropyWithLogitsGrad();

  310.   _SinPrimType_Sin();

  311.   _SmoothL1LossPrimType_SmoothL1Loss();

  312.   _SmoothL1LossGradPrimType_SmoothL1LossGrad();

  313.   _SqrtPrimType_Sqrt();

  314.   _SquarePrimType_Square();

  315.   _ZerosLikePrimType_ZerosLike();

  316.   _AssertPrimType_Assert();

  317.   _TensorListSetItemPrimType_TensorListSetItem();

  318.   _SplicePrimType_Splice();

  319.   _SparseSoftmaxCrossEntropyWithLogitsPrimType_SparseSoftmaxCrossEntropyWithLogits();

  320.   _GRUPrimType_GRU();

  321.   _SizeOpPrimType_Size();

  322.   _CustomExtractFeaturesPrimType_CustomExtractFeatures();

  323.   _CustomPredictPrimType_CustomPredict();

  324.   _WhilePrimType_While();

  325.   _StridedSliceGradPrimType_StridedSliceGrad();

  326.   _LshProjectionPrimType_LshProjection();

  327.   _SoftmaxPrimType_Softmax();

  328.   _CustomNormalizePrimType_CustomNormalize();

  329.   _UnstackPrimType_Unstack();

  330.   _ROIPoolingPrimType_ROIPooling();

  331.   _LayerNormGradPrimType_LayerNormGrad();

  332.   _DropoutGradPrimType_DropoutGrad();

  333.   _TopKPrimType_TopKFusion();

  334.   _ApplyMomentumPrimType_ApplyMomentum();

  335.   _AdderPrimType_AdderFusion();

  336.   _Conv2DPrimType_Conv2DFusion();

  337.   _UniformRealPrimType_UniformReal();

  338.   _AvgPoolGradPrimType_AvgPoolGrad();

  339.   _MaxPoolGradPrimType_MaxPoolGrad();

  340.   _DepthToSpacePrimType_DepthToSpace();

  341.   _Conv2dTransposePrimType_Conv2dTransposeFusion();

  342.   _QuantDTypeCastPrimType_QuantDTypeCast();

  343.   _FftImagPrimType_FftImag();

  344.   _FftRealPrimType_FftReal();

  345.   _ResizePrimType_Resize();

  346.   _SpaceToBatchNDPrimType_SpaceToBatchND();

  347.   _TransposePrimType_Transpose();

  348.   _TensorListReservePrimType_TensorListReserve();

  349.   _ShapePrimType_Shape();

  350.   _BiasAddGradPrimType_BiasAddGrad();

  351.   _MatMulPrimType_MatMul();

  352.   _DropoutPrimType_Dropout();

  353.   _ReshapePrimType_Reshape();

  354.   _PadPrimType_PadFusion();

  355.   _TensorListStackPrimType_TensorListStack();

  356.   _FlattenPrimType_Flatten();

  357.   _BatchNormGradPrimType_BatchNormGrad();

  358.   _SpaceToBatchPrimType_SpaceToBatch();

  359.   _GatherNdPrimType_GatherNd();

  360.   _CropAndResizePrimType_CropAndResize();

  361.   _SparseToDensePrimType_SparseToDense();

  362.   _AddPrimType_AddFusion();

  363.   _DivPrimType_DivFusion();

  364.   _EltwisePrimType_Eltwise();

  365.   _FloorDivPrimType_FloorDiv();

  366.   _FloorModPrimType_FloorMod();

  367.   _LogicalAndPrimType_LogicalAnd();

  368.   _LogicalOrPrimType_LogicalOr();

  369.   _MaximumPrimType_Maximum();

  370.   _MinimumPrimType_Minimum();

  371.   _ModPrimType_Mod();

  372.   _MulPrimType_MulFusion();

  373.   _RealDivPrimType_RealDiv();

  374.   _SubPrimType_SubFusion();

  375.   _SquaredDifferencePrimType_SquaredDifference();

  376. }

  377. #endif

  378. InferShape *g_infer_func = NULL;

  379. 

  380. __attribute__((constructor(101))) void InitInferFuncBuf() {

  381.   if (g_infer_func != NULL) {

  382.     return;

  383.   }

  384.   g_infer_func = malloc(PrimType_MAX * sizeof(InferShape));

  385.   if (g_infer_func != NULL) {

  386.     memset(g_infer_func, 0, PrimType_MAX * sizeof(InferShape));

  387.   }

  388. #ifdef MS_COMPILE_IOS

  389.   RegisterInfer();

  390. #endif

  391. }

  392. 

  393. __attribute__((destructor)) void DestroyInferFuncBuf() {

  394.   if (g_infer_func == NULL) {

  395.     return;

  396.   }

  397.   free(g_infer_func);

  398.   g_infer_func = NULL;

  399. }

  400. 

  401. InferShape GetInferFunc(int prim_type) {

  402.   if (g_infer_func != NULL && prim_type < PrimType_MAX) {

  403.     return g_infer_func[prim_type];

  404.   }

  405.   return NULL;

  406. }

  407. 

  408. void RegInfer(int prim_type, InferShape func) {

  409.   if (g_infer_func != NULL && prim_type < PrimType_MAX) {

  410.     g_infer_func[prim_type] = func;

  411.   }

  412. }