source/shape/ShapeRegister.cpp (248 lines of code) (raw):
// This file is generated by Shell for ops register
namespace MNN {
extern void ___ShapeSizeComputer__OpType_Shape__();
extern void ___ShapeRasterComputer__OpType_Raster__();
extern void ___PriorBoxComputer__OpType_PriorBox__();
extern void ___ShapeBroadcastTo__OpType_BroadcastTo__();
extern void ___InterpComputer__OpType_Interp__();
extern void ___CropSizeComputer__OpType_Crop__();
extern void ___MatMulSizeComputer__OpType_MatMul__();
extern void ___MatMulSizeComputer__OpType_BatchMatMul__();
extern void ___DynamicQuantComputer__OpType_DynamicQuant__();
extern void ___ReshapeComputer__OpType_Reshape__();
extern void ___ReshapeComputer__OpType_QuantizedReshape__();
extern void ___FlattenComputer__OpType_Flatten__();
extern void ___StridedSliceComputer__OpType_StridedSlice__();
extern void ___PoolSizeComputer__OpType_Pooling__();
extern void ___PoolSizeComputer__OpType_PoolInt8__();
extern void ___InnerProductComputer__OpType_InnerProduct__();
extern void ___ReductionComputer__OpType_Reduction__();
extern void ___QuantizedAvgPoolComputer__OpType_QuantizedAvgPool__();
extern void ___ArgMaxComputer__OpType_ArgMax__();
extern void ___ArgMaxComputer__OpType_ArgMin__();
extern void ___GridSampleSizeComputer__OpType_GridSample__();
extern void ___DepthToSpaceSizeComputer__OpType_DepthToSpace__();
extern void ___SliceTfComputer__OpType_SliceTf__();
extern void ___SelectSizeComputer__OpType_Select__();
extern void ___SegmentMeanSizeComputer__OpType_Segment__();
extern void ___DetComputer__OpType_Det__();
extern void ___ResizeComputer__OpType_Resize__();
extern void ___ImageProcessComputer__OpType_ImageProcess__();
extern void ___TransposeComputer__OpType_Transpose__();
extern void ___WhereSizeComputer__OpType_Where__();
extern void ___SqueezeSizeComputer__OpType_Squeeze__();
extern void ___UnSqueezeSizeComputer__OpType_Unsqueeze__();
extern void ___RandomUniformComputer__OpType_RandomUniform__();
extern void ___RandomUniformComputer__OpType_RandomNormal__();
extern void ___BinaryOpComputer__OpType_BinaryOp__();
extern void ___NonMaxSuppressionV2Computer__OpType_NonMaxSuppressionV2__();
extern void ___GatherV2Computer__OpType_GatherV2__();
extern void ___GatherV2Computer__OpType_Gather__();
extern void ___Convolution3DSizeComputer__OpType_Convolution3D__();
extern void ___ConcatSizeComputer__OpType_Concat__();
extern void ___ConcatSizeComputer__OpType_QuantizedConcat__();
extern void ___CastSizeComputer__OpType_Cast__();
extern void ___CastSizeComputer__OpType_CastLike__();
extern void ___CastSizeComputer__OpType_FloatToInt8__();
extern void ___CastSizeComputer__OpType_Int8ToFloat__();
extern void ___BatchToSpaceNDSizeComputer__OpType_BatchToSpaceND__();
extern void ___LinSpaceSizeComputer__OpType_LinSpace__();
extern void ___TileComputer__OpType_Tile__();
extern void ___ShapeSetDiff1D__OpType_SetDiff1D__();
extern void ___DetectionOutputComputer__OpType_DetectionOutput__();
extern void ___ROIAlignComputer__OpType_ROIAlign__();
extern void ___DetectionPostProcessSize__OpType_DetectionPostProcess__();
extern void ___SizeOpComputer__OpType_Size__();
extern void ___SizeOpComputer__OpType_Rank__();
extern void ___TensorConvertSizeComputer__OpType_ConvertTensor__();
extern void ___GatherNDComputer__OpType_GatherND__();
extern void ___GatherElementsComputer__OpType_GatherElements__();
extern void ___PermuteComputer__OpType_Permute__();
extern void ___ExpandDimsComputer__OpType_ExpandDims__();
extern void ___ShapeOneHot__OpType_OneHot__();
extern void ___ROIPoolingComputer__OpType_ROIPooling__();
extern void ___PaddingComputer__OpType_Padding__();
extern void ___ConvTranspose3DSizeComputer__OpType_ConvTranspose3D__();
extern void ___SpaceToDepthSizeComputer__OpType_SpaceToDepth__();
extern void ___UnravelIndexSize__OpType_UnravelIndex__();
extern void ___ShapeDequantize__OpType_Dequantize__();
extern void ___PluginSizeComputer__OpType_Plugin__();
extern void ___CropAndResizeComputer__OpType_CropAndResize__();
extern void ___QuantizedMaxPoolComputer__OpType_QuantizedMaxPool__();
extern void ___Pool3DSizeComputer__OpType_Pooling3D__();
extern void ___MomentsComputer__OpType_Moments__();
extern void ___RangeComputer__OpType_Range__();
extern void ___UnpackComputer__OpType_Unpack__();
extern void ___SvdComputer__OpType_Svd__();
extern void ___TopKV2SizeComputer__OpType_TopKV2__();
extern void ___ShapeUnique__OpType_Unique__();
extern void ___RNNSequenceGRUComputer__OpType_RNNSequenceGRU__();
extern void ___ShapeHistogram__OpType_Histogram__();
extern void ___FillComputer__OpType_Fill__();
extern void ___ProposalComputer__OpType_Proposal__();
extern void ___CosineSimilaritySize__OpType_CosineSimilarity__();
extern void ___TensorArrayComputer__OpType_TensorArray__();
extern void ___TensorArraySizeComputer__OpType_TensorArraySize__();
extern void ___TensorArrayReadComputer__OpType_TensorArrayRead__();
extern void ___TensorArrayWriteComputer__OpType_TensorArrayWrite__();
extern void ___TensorArrayGatherComputer__OpType_TensorArrayGather__();
extern void ___TensorArrayScatterComputer__OpType_TensorArrayScatter__();
extern void ___TensorArraySplitComputer__OpType_TensorArraySplit__();
extern void ___TensorArrayConcatComputer__OpType_TensorArrayConcat__();
extern void ___TensorArrayInsertComputer__OpType_TensorArrayInsert__();
extern void ___TensorArrayEraseComputer__OpType_TensorArrayErase__();
extern void ___ConvolutionSizeComputer__OpType_Convolution__();
extern void ___ConvolutionSizeComputer__OpType_ConvolutionDepthwise__();
extern void ___ConvolutionSizeComputer__OpType_TfQuantizedConv2D__();
extern void ___ConvolutionSizeComputer__OpType_QuantizedDepthwiseConv2D__();
extern void ___ConvolutionSizeComputer__OpType_ConvInt8__();
extern void ___ConvolutionSizeComputer__OpType_DepthwiseConvInt8__();
extern void ___Dilation2DSizeComputer__OpType_Dilation2D__();
extern void ___Conv2DBackpropFilterSizeComputer__OpType_Conv2DBackPropFilter__();
extern void ___Im2ColSizeComputer__OpType_Im2Col__();
extern void ___Col2ImSizeComputer__OpType_Col2Im__();
extern void ___ShapeScatterNd__OpType_ScatterNd__();
extern void ___StftOpComputer__OpType_Stft__();
extern void ___LSTMComputer__OpType_LSTM__();
extern void ___LSTMBlockCellComputer__OpType_LSTMBlockCell__();
extern void ___RNNComputer__OpType_RNN__();
extern void ___SliceComputer__OpType_Slice__();
extern void ___SpaceToBatchNDSizeComputer__OpType_SpaceToBatchND__();
extern void ___PackComputer__OpType_Pack__();
extern void ___DeconvolutionSizeComputer__OpType_Deconvolution__();
extern void ___DeconvolutionSizeComputer__OpType_DeconvolutionDepthwise__();
#ifdef MNN_SUPPORT_RENDER
extern void ___RasterAndInterpolateComputer__OpType_RasterAndInterpolate__();
extern void ___TextureComputer__OpType_Texture__();
#endif
#ifdef MNN_SUPPORT_TRANSFORMER_FUSE
extern void ___SplitGeLUSizeComputer__OpType_SplitGeLU__();
extern void ___SeqLen2SpatialSizeComputer__OpType_SeqLen2Spatial__();
extern void ___FmhaV2SizeComputer__OpType_FmhaV2__();
extern void ___FmhcaSizeComputer__OpType_Fmhca__();
extern void ___AttentionSizeComputer__OpType_Attention__();
#endif
void registerShapeOps() {
___ShapeSizeComputer__OpType_Shape__();
___ShapeRasterComputer__OpType_Raster__();
___PriorBoxComputer__OpType_PriorBox__();
___ShapeBroadcastTo__OpType_BroadcastTo__();
___InterpComputer__OpType_Interp__();
___CropSizeComputer__OpType_Crop__();
___MatMulSizeComputer__OpType_MatMul__();
___MatMulSizeComputer__OpType_BatchMatMul__();
___DynamicQuantComputer__OpType_DynamicQuant__();
___ReshapeComputer__OpType_Reshape__();
___ReshapeComputer__OpType_QuantizedReshape__();
___FlattenComputer__OpType_Flatten__();
___StridedSliceComputer__OpType_StridedSlice__();
___PoolSizeComputer__OpType_Pooling__();
___PoolSizeComputer__OpType_PoolInt8__();
___InnerProductComputer__OpType_InnerProduct__();
___ReductionComputer__OpType_Reduction__();
___QuantizedAvgPoolComputer__OpType_QuantizedAvgPool__();
___ArgMaxComputer__OpType_ArgMax__();
___ArgMaxComputer__OpType_ArgMin__();
___GridSampleSizeComputer__OpType_GridSample__();
___DepthToSpaceSizeComputer__OpType_DepthToSpace__();
___SliceTfComputer__OpType_SliceTf__();
___SelectSizeComputer__OpType_Select__();
___SegmentMeanSizeComputer__OpType_Segment__();
___DetComputer__OpType_Det__();
___ResizeComputer__OpType_Resize__();
___ImageProcessComputer__OpType_ImageProcess__();
___TransposeComputer__OpType_Transpose__();
___WhereSizeComputer__OpType_Where__();
___SqueezeSizeComputer__OpType_Squeeze__();
___UnSqueezeSizeComputer__OpType_Unsqueeze__();
___RandomUniformComputer__OpType_RandomUniform__();
___RandomUniformComputer__OpType_RandomNormal__();
___BinaryOpComputer__OpType_BinaryOp__();
___NonMaxSuppressionV2Computer__OpType_NonMaxSuppressionV2__();
___GatherV2Computer__OpType_GatherV2__();
___GatherV2Computer__OpType_Gather__();
___Convolution3DSizeComputer__OpType_Convolution3D__();
___ConcatSizeComputer__OpType_Concat__();
___ConcatSizeComputer__OpType_QuantizedConcat__();
___CastSizeComputer__OpType_Cast__();
___CastSizeComputer__OpType_CastLike__();
___CastSizeComputer__OpType_FloatToInt8__();
___CastSizeComputer__OpType_Int8ToFloat__();
___BatchToSpaceNDSizeComputer__OpType_BatchToSpaceND__();
___LinSpaceSizeComputer__OpType_LinSpace__();
___TileComputer__OpType_Tile__();
___ShapeSetDiff1D__OpType_SetDiff1D__();
___DetectionOutputComputer__OpType_DetectionOutput__();
___ROIAlignComputer__OpType_ROIAlign__();
___DetectionPostProcessSize__OpType_DetectionPostProcess__();
___SizeOpComputer__OpType_Size__();
___SizeOpComputer__OpType_Rank__();
___TensorConvertSizeComputer__OpType_ConvertTensor__();
___GatherNDComputer__OpType_GatherND__();
___GatherElementsComputer__OpType_GatherElements__();
___PermuteComputer__OpType_Permute__();
___ExpandDimsComputer__OpType_ExpandDims__();
___ShapeOneHot__OpType_OneHot__();
___ROIPoolingComputer__OpType_ROIPooling__();
___PaddingComputer__OpType_Padding__();
___ConvTranspose3DSizeComputer__OpType_ConvTranspose3D__();
___SpaceToDepthSizeComputer__OpType_SpaceToDepth__();
___UnravelIndexSize__OpType_UnravelIndex__();
___ShapeDequantize__OpType_Dequantize__();
___PluginSizeComputer__OpType_Plugin__();
___CropAndResizeComputer__OpType_CropAndResize__();
___QuantizedMaxPoolComputer__OpType_QuantizedMaxPool__();
___Pool3DSizeComputer__OpType_Pooling3D__();
___MomentsComputer__OpType_Moments__();
___RangeComputer__OpType_Range__();
___UnpackComputer__OpType_Unpack__();
___SvdComputer__OpType_Svd__();
___TopKV2SizeComputer__OpType_TopKV2__();
___ShapeUnique__OpType_Unique__();
___RNNSequenceGRUComputer__OpType_RNNSequenceGRU__();
___ShapeHistogram__OpType_Histogram__();
___FillComputer__OpType_Fill__();
___ProposalComputer__OpType_Proposal__();
___CosineSimilaritySize__OpType_CosineSimilarity__();
___TensorArrayComputer__OpType_TensorArray__();
___TensorArraySizeComputer__OpType_TensorArraySize__();
___TensorArrayReadComputer__OpType_TensorArrayRead__();
___TensorArrayWriteComputer__OpType_TensorArrayWrite__();
___TensorArrayGatherComputer__OpType_TensorArrayGather__();
___TensorArrayScatterComputer__OpType_TensorArrayScatter__();
___TensorArraySplitComputer__OpType_TensorArraySplit__();
___TensorArrayConcatComputer__OpType_TensorArrayConcat__();
___TensorArrayInsertComputer__OpType_TensorArrayInsert__();
___TensorArrayEraseComputer__OpType_TensorArrayErase__();
___ConvolutionSizeComputer__OpType_Convolution__();
___ConvolutionSizeComputer__OpType_ConvolutionDepthwise__();
___ConvolutionSizeComputer__OpType_TfQuantizedConv2D__();
___ConvolutionSizeComputer__OpType_QuantizedDepthwiseConv2D__();
___ConvolutionSizeComputer__OpType_ConvInt8__();
___ConvolutionSizeComputer__OpType_DepthwiseConvInt8__();
___Dilation2DSizeComputer__OpType_Dilation2D__();
___Conv2DBackpropFilterSizeComputer__OpType_Conv2DBackPropFilter__();
___Im2ColSizeComputer__OpType_Im2Col__();
___Col2ImSizeComputer__OpType_Col2Im__();
___ShapeScatterNd__OpType_ScatterNd__();
___StftOpComputer__OpType_Stft__();
___LSTMComputer__OpType_LSTM__();
___LSTMBlockCellComputer__OpType_LSTMBlockCell__();
___RNNComputer__OpType_RNN__();
___SliceComputer__OpType_Slice__();
___SpaceToBatchNDSizeComputer__OpType_SpaceToBatchND__();
___PackComputer__OpType_Pack__();
___DeconvolutionSizeComputer__OpType_Deconvolution__();
___DeconvolutionSizeComputer__OpType_DeconvolutionDepthwise__();
#ifdef MNN_SUPPORT_RENDER
___RasterAndInterpolateComputer__OpType_RasterAndInterpolate__();
___TextureComputer__OpType_Texture__();
#endif
#ifdef MNN_SUPPORT_TRANSFORMER_FUSE
___SplitGeLUSizeComputer__OpType_SplitGeLU__();
___SeqLen2SpatialSizeComputer__OpType_SeqLen2Spatial__();
___FmhaV2SizeComputer__OpType_FmhaV2__();
___FmhcaSizeComputer__OpType_Fmhca__();
___AttentionSizeComputer__OpType_Attention__();
#endif
}
}