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 } }