MNN/source/backend/cpu/CPUOPRegister.cpp

229 lines
12 KiB
C++
Raw Normal View History

2019-05-08 16:37:43 +08:00
// This file is generated by Shell for ops register
namespace MNN {
2019-05-09 19:39:33 +08:00
#ifdef MNN_CODEGEN_REGISTER
extern void ___CPUArgMaxCreator__OpType_ArgMax__();
2019-12-27 22:16:57 +08:00
extern void ___CPUArgMaxCreator__OpType_ArgMin__();
extern void ___CPUAsStringCreator__OpType_AsString__();
extern void ___CPUBatchMatMulCreator__OpType_BatchMatMul__();
extern void ___CPUBatchToSpaceNDCreator__OpType_BatchToSpaceND__();
extern void ___CPUBinaryCreator__OpType_BinaryOp__();
2019-12-27 22:16:57 +08:00
extern void ___CPUBroadcastToCreator__OpType_BroadcastTo__();
extern void ___CPUCastCreator__OpType_Cast__();
extern void ___CPUConcatCreator__OpType_Concat__();
extern void ___CPUConstCreator__OpType_Const__();
2019-12-27 22:16:57 +08:00
extern void ___CPUConstCreator__OpType_TrainableParam__();
extern void ___CPUConv2DBackPropFilterCreator__OpType_Conv2DBackPropFilter__();
extern void ___CPUConvInt8Creator__OpType_ConvInt8__();
extern void ___ConvolutionFactory__OpType_Convolution__();
extern void ___Convolution3DCreator__OpType_Convolution3D__();
extern void ___CPUConvolutionDepthwiseCreator__OpType_ConvolutionDepthwise__();
extern void ___CPUCosineSimilarityCreator__OpType_CosineSimilarity__();
extern void ___CPUCropCreator__OpType_Crop__();
extern void ___CPUCropAndResizeCreator__OpType_CropAndResize__();
extern void ___CPUDeconvolutionCreator__OpType_Deconvolution__();
extern void ___CPUDeconvolutionDepthwiseCreator__OpType_DeconvolutionDepthwise__();
extern void ___DepthToSpaceCreator__OpType_DepthToSpace__();
extern void ___CPUDepthwiseConvInt8Creator__OpType_DepthwiseConvInt8__();
extern void ___CPUDequantizeCreator__OpType_Dequantize__();
extern void ___CPUDetectionOutputCreator__OpType_DetectionOutput__();
extern void ___CPUDetectionPostProcessCreator__OpType_DetectionPostProcess__();
2019-12-27 22:16:57 +08:00
extern void ___CPUDilation2DCreator__OpType_Dilation2D__();
extern void ___CPUEltwiesCreator__OpType_Eltwise__();
extern void ___CPUEltwiseInt8Creator__OpType_EltwiseInt8__();
extern void ___CPUEluCreator__OpType_ELU__();
extern void ___CPUExpandDimsCreator__OpType_ExpandDims__();
extern void ___CPUFillCreator__OpType_Fill__();
extern void ___CPUFloatToInt8Creator__OpType_FloatToInt8__();
extern void ___CPUGatherCreator__OpType_Gather__();
- build: - unify schema building in core and converter; - add more build script for android; - add linux build script for python; - ops impl: - add floor mod support in binary; - use eltwise impl in add/max/sub/mul binary for optimization; - remove fake double support in cast; - fix 5d support for concat; - add adjX and adjY support for batch matmul; - optimize conv2d back prop filter; - add pad mode support for conv3d; - fix bug in conv2d & conv depthwise with very small feature map; - optimize binary without broacast; - add data types support for gather; - add gather ND support; - use uint8 data type in gather v2; - add transpose support for matmul; - add matrix band part; - add dim != 4 support for padding, reshape & tensor convert; - add pad type support for pool3d; - make ops based on TensorFlow Lite quantization optional; - add all & any support for reduction; - use type in parameter as output type in reduction; - add int support for unary; - add variable weight support for conv2d; - fix conv2d depthwise weights initialization; - fix type support for transpose; - fix grad outputs count for reduce grad and reshape grad; - fix priorbox & detection output; - fix metal softmax error; - python: - add runSessionWithCallBackInfo interface; - add max nodes limit (1400) for visualization tool; - fix save error in python3; - align default dim; - convert: - add extra design for optimization; - add more post converting optimizers; - add caffe v1 weights blob support; - add cast, unary, conv transpose support for onnx model; - optimize batchnorm, conv with variable weights, prelu, reshape, slice, upsample for onnx model; - add cos/sin/atan/tan support for unary for tensorflow model; - add any/all support for reduction for tensorflow model; - add elu, conv3d, pool3d support for tensorflow model; - optimize argmax, batchnorm, concat, batch to space, conv with variable weights, prelu, slice for tensorflow model; - others: - fix size computer lock; - fix thread pool deadlock; - add express & parameters in express; - rewrite blitter chooser without static map; - add tests for expr;
2019-10-29 13:37:26 +08:00
extern void ___CPUGatherNDCreator__OpType_GatherND__();
extern void ___CPUGatherV2Creator__OpType_GatherV2__();
extern void ___CPUInnerProductCreator__OpType_InnerProduct__();
2019-12-27 22:16:57 +08:00
extern void ___CPUInstanceNormCreator__OpType_InstanceNorm__();
extern void ___CPUInt8ToFloatCreator__OpType_Int8ToFloat__();
extern void ___CPUInterpCreator__OpType_Interp__();
extern void ___CPULRNCreator__OpType_LRN__();
extern void ___CPULSTMCreator__OpType_LSTM__();
2019-12-27 22:16:57 +08:00
extern void ___CPULinSpaceCreator__OpType_LinSpace__();
extern void ___CPUMatMulCreator__OpType_MatMul__();
- build: - unify schema building in core and converter; - add more build script for android; - add linux build script for python; - ops impl: - add floor mod support in binary; - use eltwise impl in add/max/sub/mul binary for optimization; - remove fake double support in cast; - fix 5d support for concat; - add adjX and adjY support for batch matmul; - optimize conv2d back prop filter; - add pad mode support for conv3d; - fix bug in conv2d & conv depthwise with very small feature map; - optimize binary without broacast; - add data types support for gather; - add gather ND support; - use uint8 data type in gather v2; - add transpose support for matmul; - add matrix band part; - add dim != 4 support for padding, reshape & tensor convert; - add pad type support for pool3d; - make ops based on TensorFlow Lite quantization optional; - add all & any support for reduction; - use type in parameter as output type in reduction; - add int support for unary; - add variable weight support for conv2d; - fix conv2d depthwise weights initialization; - fix type support for transpose; - fix grad outputs count for reduce grad and reshape grad; - fix priorbox & detection output; - fix metal softmax error; - python: - add runSessionWithCallBackInfo interface; - add max nodes limit (1400) for visualization tool; - fix save error in python3; - align default dim; - convert: - add extra design for optimization; - add more post converting optimizers; - add caffe v1 weights blob support; - add cast, unary, conv transpose support for onnx model; - optimize batchnorm, conv with variable weights, prelu, reshape, slice, upsample for onnx model; - add cos/sin/atan/tan support for unary for tensorflow model; - add any/all support for reduction for tensorflow model; - add elu, conv3d, pool3d support for tensorflow model; - optimize argmax, batchnorm, concat, batch to space, conv with variable weights, prelu, slice for tensorflow model; - others: - fix size computer lock; - fix thread pool deadlock; - add express & parameters in express; - rewrite blitter chooser without static map; - add tests for expr;
2019-10-29 13:37:26 +08:00
extern void ___CPUMatrixBandPartCreator__OpType_MatrixBandPart__();
extern void ___CPUMomentsCreator__OpType_Moments__();
extern void ___CPUNonMaxSuppressionV2Creator__OpType_NonMaxSuppressionV2__();
extern void ___CPUNormalizeCreator__OpType_Normalize__();
2019-12-27 22:16:57 +08:00
extern void ___CPUOneHotCreator__OpType_OneHot__();
extern void ___CPUPackCreator__OpType_Pack__();
extern void ___CPUPaddingCreator__OpType_Padding__();
extern void ___CPUPermuteCreator__OpType_Permute__();
extern void ___CPUPoolCreator__OpType_Pooling__();
extern void ___CPUPool3DCreator__OpType_Pooling3D__();
extern void ___CPUPoolGradCreator__OpType_PoolGrad__();
extern void ___CPUPoolInt8Creator__OpType_PoolInt8__();
extern void ___CPUPriorBoxCreator__OpType_PriorBox__();
extern void ___CPUProposalCreator__OpType_Proposal__();
extern void ___CPUDepthwiseCreator__OpType_QuantizedDepthwiseConv2D__();
extern void ___CPUQuantizedAddCreator__OpType_QuantizedAdd__();
extern void ___CPUQuantizedAvgPoolCreator__OpType_QuantizedAvgPool__();
extern void ___CPUQuantizedConcatCreator__OpType_QuantizedConcat__();
extern void ___CPUQuantizedLogisticCreator__OpType_QuantizedLogistic__();
extern void ___CPUQuantizedMaxPoolCreator__OpType_QuantizedMaxPool__();
extern void ___CPUQuantizedReshapeCreator__OpType_QuantizedReshape__();
extern void ___CPUQuantizedSoftmaxCreator__OpType_QuantizedSoftmax__();
extern void ___CPURNNSequenceGRUCreator__OpType_RNNSequenceGRU__();
extern void ___CPUROIPoolingCreator__OpType_ROIPooling__();
extern void ___CPURangeCreator__OpType_Range__();
extern void ___CPURankCreator__OpType_Rank__();
extern void ___CPUReduceJoinCreator__OpType_ReduceJoin__();
extern void ___CPUReductionCreator__OpType_Reduction__();
extern void ___CPUReluCreator__OpType_ReLU__();
extern void ___CPUReluCreator__OpType_PReLU__();
extern void ___CPURelu6Creator__OpType_ReLU6__();
extern void ___CPUReluGradCreator__OpType_ReluGrad__();
extern void ___CPUReluGradCreator__OpType_Relu6Grad__();
extern void ___CPUReshapeCreator__OpType_Reshape__();
extern void ___CPUResizeCreator__OpType_Resize__();
extern void ___CPUReverseSequenceCreator__OpType_ReverseSequence__();
extern void ___CPUScaleCreator__OpType_Scale__();
2019-12-27 22:16:57 +08:00
extern void ___CPUScatterNdCreator__OpType_ScatterNd__();
extern void ___CPUSelectCreator__OpType_Select__();
extern void ___CPUSeluCreator__OpType_Selu__();
extern void ___CPUSetDiff1DCreator__OpType_SetDiff1D__();
extern void ___CPUShapeCreator__OpType_Shape__();
extern void ___CPUSigmoidCreator__OpType_Sigmoid__();
extern void ___CPUSizeCreator__OpType_Size__();
extern void ___CPUSliceCreator__OpType_Slice__();
extern void ___CPUSliceTfCreator__OpType_SliceTf__();
extern void ___CPUSoftmaxCreator__OpType_Softmax__();
extern void ___CPUSoftmaxGradCreator__OpType_SoftmaxGrad__();
extern void ___SpaceBatchCreator__OpType_SpaceToBatchND__();
extern void ___SpaceToDepthCreator__OpType_SpaceToDepth__();
extern void ___CPUSpatialProductCreator__OpType_SpatialProduct__();
extern void ___CPUSqueezeCreator__OpType_Squeeze__();
extern void ___CPUSqueezeCreator__OpType_Unsqueeze__();
extern void ___CPUStridedSliceCreator__OpType_StridedSlice__();
extern void ___CPUTFQuantizedConv2DCreator__OpType_TfQuantizedConv2D__();
extern void ___CPUTanhCreator__OpType_TanH__();
extern void ___CPUTensorConvertFactory__OpType_ConvertTensor__();
2019-12-27 22:16:57 +08:00
extern void ___CPUThresholdCreator__OpType_Threshold__();
extern void ___CPUTileCreator__OpType_Tile__();
extern void ___CPUTopKV2Creator__OpType_TopKV2__();
extern void ___CPUTransposeeCreator__OpType_Transpose__();
extern void ___CPUUnaryCreator__OpType_UnaryOp__();
extern void ___CPUUnpackCreator__OpType_Unpack__();
2019-12-27 22:16:57 +08:00
extern void ___CPUUnravelIndexCreator__OpType_UnravelIndex__();
extern void ___CPUWhereCreator__OpType_Where__();
extern void ___CPUZeroLikeCreator__OpType_ZerosLike__();
2019-05-08 16:37:43 +08:00
void registerCPUOps() {
___CPUArgMaxCreator__OpType_ArgMax__();
2019-12-27 22:16:57 +08:00
___CPUArgMaxCreator__OpType_ArgMin__();
___CPUAsStringCreator__OpType_AsString__();
___CPUBatchMatMulCreator__OpType_BatchMatMul__();
___CPUBatchToSpaceNDCreator__OpType_BatchToSpaceND__();
___CPUBinaryCreator__OpType_BinaryOp__();
2019-12-27 22:16:57 +08:00
___CPUBroadcastToCreator__OpType_BroadcastTo__();
___CPUCastCreator__OpType_Cast__();
___CPUConcatCreator__OpType_Concat__();
___CPUConstCreator__OpType_Const__();
2019-12-27 22:16:57 +08:00
___CPUConstCreator__OpType_TrainableParam__();
___CPUConv2DBackPropFilterCreator__OpType_Conv2DBackPropFilter__();
___CPUConvInt8Creator__OpType_ConvInt8__();
___ConvolutionFactory__OpType_Convolution__();
___Convolution3DCreator__OpType_Convolution3D__();
___CPUConvolutionDepthwiseCreator__OpType_ConvolutionDepthwise__();
___CPUCosineSimilarityCreator__OpType_CosineSimilarity__();
___CPUCropCreator__OpType_Crop__();
___CPUCropAndResizeCreator__OpType_CropAndResize__();
___CPUDeconvolutionCreator__OpType_Deconvolution__();
___CPUDeconvolutionDepthwiseCreator__OpType_DeconvolutionDepthwise__();
___DepthToSpaceCreator__OpType_DepthToSpace__();
___CPUDepthwiseConvInt8Creator__OpType_DepthwiseConvInt8__();
___CPUDequantizeCreator__OpType_Dequantize__();
___CPUDetectionOutputCreator__OpType_DetectionOutput__();
___CPUDetectionPostProcessCreator__OpType_DetectionPostProcess__();
2019-12-27 22:16:57 +08:00
___CPUDilation2DCreator__OpType_Dilation2D__();
___CPUEltwiesCreator__OpType_Eltwise__();
___CPUEltwiseInt8Creator__OpType_EltwiseInt8__();
___CPUEluCreator__OpType_ELU__();
___CPUExpandDimsCreator__OpType_ExpandDims__();
___CPUFillCreator__OpType_Fill__();
___CPUFloatToInt8Creator__OpType_FloatToInt8__();
___CPUGatherCreator__OpType_Gather__();
- build: - unify schema building in core and converter; - add more build script for android; - add linux build script for python; - ops impl: - add floor mod support in binary; - use eltwise impl in add/max/sub/mul binary for optimization; - remove fake double support in cast; - fix 5d support for concat; - add adjX and adjY support for batch matmul; - optimize conv2d back prop filter; - add pad mode support for conv3d; - fix bug in conv2d & conv depthwise with very small feature map; - optimize binary without broacast; - add data types support for gather; - add gather ND support; - use uint8 data type in gather v2; - add transpose support for matmul; - add matrix band part; - add dim != 4 support for padding, reshape & tensor convert; - add pad type support for pool3d; - make ops based on TensorFlow Lite quantization optional; - add all & any support for reduction; - use type in parameter as output type in reduction; - add int support for unary; - add variable weight support for conv2d; - fix conv2d depthwise weights initialization; - fix type support for transpose; - fix grad outputs count for reduce grad and reshape grad; - fix priorbox & detection output; - fix metal softmax error; - python: - add runSessionWithCallBackInfo interface; - add max nodes limit (1400) for visualization tool; - fix save error in python3; - align default dim; - convert: - add extra design for optimization; - add more post converting optimizers; - add caffe v1 weights blob support; - add cast, unary, conv transpose support for onnx model; - optimize batchnorm, conv with variable weights, prelu, reshape, slice, upsample for onnx model; - add cos/sin/atan/tan support for unary for tensorflow model; - add any/all support for reduction for tensorflow model; - add elu, conv3d, pool3d support for tensorflow model; - optimize argmax, batchnorm, concat, batch to space, conv with variable weights, prelu, slice for tensorflow model; - others: - fix size computer lock; - fix thread pool deadlock; - add express & parameters in express; - rewrite blitter chooser without static map; - add tests for expr;
2019-10-29 13:37:26 +08:00
___CPUGatherNDCreator__OpType_GatherND__();
___CPUGatherV2Creator__OpType_GatherV2__();
___CPUInnerProductCreator__OpType_InnerProduct__();
2019-12-27 22:16:57 +08:00
___CPUInstanceNormCreator__OpType_InstanceNorm__();
___CPUInt8ToFloatCreator__OpType_Int8ToFloat__();
___CPUInterpCreator__OpType_Interp__();
___CPULRNCreator__OpType_LRN__();
___CPULSTMCreator__OpType_LSTM__();
2019-12-27 22:16:57 +08:00
___CPULinSpaceCreator__OpType_LinSpace__();
___CPUMatMulCreator__OpType_MatMul__();
- build: - unify schema building in core and converter; - add more build script for android; - add linux build script for python; - ops impl: - add floor mod support in binary; - use eltwise impl in add/max/sub/mul binary for optimization; - remove fake double support in cast; - fix 5d support for concat; - add adjX and adjY support for batch matmul; - optimize conv2d back prop filter; - add pad mode support for conv3d; - fix bug in conv2d & conv depthwise with very small feature map; - optimize binary without broacast; - add data types support for gather; - add gather ND support; - use uint8 data type in gather v2; - add transpose support for matmul; - add matrix band part; - add dim != 4 support for padding, reshape & tensor convert; - add pad type support for pool3d; - make ops based on TensorFlow Lite quantization optional; - add all & any support for reduction; - use type in parameter as output type in reduction; - add int support for unary; - add variable weight support for conv2d; - fix conv2d depthwise weights initialization; - fix type support for transpose; - fix grad outputs count for reduce grad and reshape grad; - fix priorbox & detection output; - fix metal softmax error; - python: - add runSessionWithCallBackInfo interface; - add max nodes limit (1400) for visualization tool; - fix save error in python3; - align default dim; - convert: - add extra design for optimization; - add more post converting optimizers; - add caffe v1 weights blob support; - add cast, unary, conv transpose support for onnx model; - optimize batchnorm, conv with variable weights, prelu, reshape, slice, upsample for onnx model; - add cos/sin/atan/tan support for unary for tensorflow model; - add any/all support for reduction for tensorflow model; - add elu, conv3d, pool3d support for tensorflow model; - optimize argmax, batchnorm, concat, batch to space, conv with variable weights, prelu, slice for tensorflow model; - others: - fix size computer lock; - fix thread pool deadlock; - add express & parameters in express; - rewrite blitter chooser without static map; - add tests for expr;
2019-10-29 13:37:26 +08:00
___CPUMatrixBandPartCreator__OpType_MatrixBandPart__();
___CPUMomentsCreator__OpType_Moments__();
___CPUNonMaxSuppressionV2Creator__OpType_NonMaxSuppressionV2__();
___CPUNormalizeCreator__OpType_Normalize__();
2019-12-27 22:16:57 +08:00
___CPUOneHotCreator__OpType_OneHot__();
___CPUPackCreator__OpType_Pack__();
___CPUPaddingCreator__OpType_Padding__();
___CPUPermuteCreator__OpType_Permute__();
___CPUPoolCreator__OpType_Pooling__();
___CPUPool3DCreator__OpType_Pooling3D__();
___CPUPoolGradCreator__OpType_PoolGrad__();
___CPUPoolInt8Creator__OpType_PoolInt8__();
___CPUPriorBoxCreator__OpType_PriorBox__();
___CPUProposalCreator__OpType_Proposal__();
___CPUDepthwiseCreator__OpType_QuantizedDepthwiseConv2D__();
___CPUQuantizedAddCreator__OpType_QuantizedAdd__();
___CPUQuantizedAvgPoolCreator__OpType_QuantizedAvgPool__();
___CPUQuantizedConcatCreator__OpType_QuantizedConcat__();
___CPUQuantizedLogisticCreator__OpType_QuantizedLogistic__();
___CPUQuantizedMaxPoolCreator__OpType_QuantizedMaxPool__();
___CPUQuantizedReshapeCreator__OpType_QuantizedReshape__();
___CPUQuantizedSoftmaxCreator__OpType_QuantizedSoftmax__();
___CPURNNSequenceGRUCreator__OpType_RNNSequenceGRU__();
___CPUROIPoolingCreator__OpType_ROIPooling__();
___CPURangeCreator__OpType_Range__();
___CPURankCreator__OpType_Rank__();
___CPUReduceJoinCreator__OpType_ReduceJoin__();
___CPUReductionCreator__OpType_Reduction__();
___CPUReluCreator__OpType_ReLU__();
___CPUReluCreator__OpType_PReLU__();
___CPURelu6Creator__OpType_ReLU6__();
___CPUReluGradCreator__OpType_ReluGrad__();
___CPUReluGradCreator__OpType_Relu6Grad__();
___CPUReshapeCreator__OpType_Reshape__();
___CPUResizeCreator__OpType_Resize__();
___CPUReverseSequenceCreator__OpType_ReverseSequence__();
___CPUScaleCreator__OpType_Scale__();
2019-12-27 22:16:57 +08:00
___CPUScatterNdCreator__OpType_ScatterNd__();
___CPUSelectCreator__OpType_Select__();
___CPUSeluCreator__OpType_Selu__();
___CPUSetDiff1DCreator__OpType_SetDiff1D__();
___CPUShapeCreator__OpType_Shape__();
___CPUSigmoidCreator__OpType_Sigmoid__();
___CPUSizeCreator__OpType_Size__();
___CPUSliceCreator__OpType_Slice__();
___CPUSliceTfCreator__OpType_SliceTf__();
___CPUSoftmaxCreator__OpType_Softmax__();
___CPUSoftmaxGradCreator__OpType_SoftmaxGrad__();
___SpaceBatchCreator__OpType_SpaceToBatchND__();
___SpaceToDepthCreator__OpType_SpaceToDepth__();
___CPUSpatialProductCreator__OpType_SpatialProduct__();
___CPUSqueezeCreator__OpType_Squeeze__();
___CPUSqueezeCreator__OpType_Unsqueeze__();
___CPUStridedSliceCreator__OpType_StridedSlice__();
___CPUTFQuantizedConv2DCreator__OpType_TfQuantizedConv2D__();
___CPUTanhCreator__OpType_TanH__();
___CPUTensorConvertFactory__OpType_ConvertTensor__();
2019-12-27 22:16:57 +08:00
___CPUThresholdCreator__OpType_Threshold__();
___CPUTileCreator__OpType_Tile__();
___CPUTopKV2Creator__OpType_TopKV2__();
___CPUTransposeeCreator__OpType_Transpose__();
___CPUUnaryCreator__OpType_UnaryOp__();
___CPUUnpackCreator__OpType_Unpack__();
2019-12-27 22:16:57 +08:00
___CPUUnravelIndexCreator__OpType_UnravelIndex__();
___CPUWhereCreator__OpType_Where__();
___CPUZeroLikeCreator__OpType_ZerosLike__();
}
2019-05-09 19:39:33 +08:00
#endif
}