for add convolutionv2 param
GitOrigin-RevId: 9d0cf55ef4
tags/v1.3.0
| @@ -65,7 +65,7 @@ decl_opr('MaskConvolution', | |||||
| params=[('param', 'Convolution')], | params=[('param', 'Convolution')], | ||||
| desc=('batched mask conv on channeled 2D images, mask is correspoding ' | desc=('batched mask conv on channeled 2D images, mask is correspoding ' | ||||
| 'to output'), | 'to output'), | ||||
| version=1) | |||||
| version=2) | |||||
| decl_opr('MaskPropagate', | decl_opr('MaskPropagate', | ||||
| inputs=[Doc('src', | inputs=[Doc('src', | ||||
| @@ -102,10 +102,11 @@ decl_opr('Local', | |||||
| 'convolution kernel in ' | 'convolution kernel in ' | ||||
| '(out row, out col, in channel, ' | '(out row, out col, in channel, ' | ||||
| 'kern row, kern col, out channel) format')], | 'kern row, kern col, out channel) format')], | ||||
| params='ConvolutionV0', | |||||
| params='Convolution', | |||||
| desc='batched convolution on channeled 2D images, but kernels are ' | desc='batched convolution on channeled 2D images, but kernels are ' | ||||
| 'not shared across different output positions') | |||||
| 'not shared across different output positions', version=2) | |||||
| """ | |||||
| decl_opr('Local', | decl_opr('Local', | ||||
| pyname='local_v1', | pyname='local_v1', | ||||
| inputs=[Doc('src', | inputs=[Doc('src', | ||||
| @@ -118,6 +119,7 @@ decl_opr('Local', | |||||
| desc='batched convolution on channeled 2D images, but kernels are ' | desc='batched convolution on channeled 2D images, but kernels are ' | ||||
| 'not shared across different output positions', | 'not shared across different output positions', | ||||
| version=1) | version=1) | ||||
| """ | |||||
| decl_opr('GroupLocal', | decl_opr('GroupLocal', | ||||
| inputs=[Doc('src', | inputs=[Doc('src', | ||||
| @@ -129,7 +131,7 @@ decl_opr('GroupLocal', | |||||
| params=[('param', 'Convolution')], | params=[('param', 'Convolution')], | ||||
| desc='batched convolution on groupped channeled 2D images, but ' | desc='batched convolution on groupped channeled 2D images, but ' | ||||
| 'kernels are not shared across different output positions', | 'kernels are not shared across different output positions', | ||||
| version=1) | |||||
| version=2) | |||||
| decl_opr('LRN', | decl_opr('LRN', | ||||
| inputs=['src'], | inputs=['src'], | ||||
| @@ -280,7 +282,7 @@ decl_opr('DeformableConvForward', | |||||
| Doc('mask', 'deformable mask in (batch, deformable group * filter row * filter col, output row, output col) format')], | Doc('mask', 'deformable mask in (batch, deformable group * filter row * filter col, output row, output col) format')], | ||||
| params=[('param', 'Convolution'), ('execution_policy', 'ExecutionPolicy')], | params=[('param', 'Convolution'), ('execution_policy', 'ExecutionPolicy')], | ||||
| desc=Doc(None, r""" Apply a deformable convolution to input tensor and filter tensor. The offset tensor will adjust the position of each grid of a convolution filter. The mask tensor will be applied to the deformed input tensor. """), | desc=Doc(None, r""" Apply a deformable convolution to input tensor and filter tensor. The offset tensor will adjust the position of each grid of a convolution filter. The mask tensor will be applied to the deformed input tensor. """), | ||||
| has_out_dtype=True) | |||||
| has_out_dtype=True, version=1) | |||||
| decl_opr('DeformablePSROIPoolingForward', | decl_opr('DeformablePSROIPoolingForward', | ||||
| pyname='deformable_psroi_pooling', | pyname='deformable_psroi_pooling', | ||||
| @@ -514,19 +514,19 @@ MGB_SEREG_OPR(ConvolutionBackwardFilterV2, 0); | |||||
| MGB_SEREG_OPR(Images2Neibs, 1); | MGB_SEREG_OPR(Images2Neibs, 1); | ||||
| MGB_SEREG_OPR(Images2NeibsBackward, 2); | MGB_SEREG_OPR(Images2NeibsBackward, 2); | ||||
| using LocalV1 = Local; | |||||
| using LocalBackwardDataV1 = LocalBackwardData; | |||||
| using LocalBackwardFilterV1 = LocalBackwardFilter; | |||||
| MGB_SEREG_OPR(LocalV1, 2); | |||||
| MGB_SEREG_OPR(LocalBackwardDataV1, 3); | |||||
| MGB_SEREG_OPR(LocalBackwardFilterV1, 3); | |||||
| using GroupLocalV1 = GroupLocal; | |||||
| using GroupLocalBackwardDataV1 = GroupLocalBackwardData; | |||||
| using GroupLocalBackwardFilterV1 = GroupLocalBackwardFilter; | |||||
| MGB_SEREG_OPR(GroupLocalV1, 2); | |||||
| MGB_SEREG_OPR(GroupLocalBackwardDataV1, 3); | |||||
| MGB_SEREG_OPR(GroupLocalBackwardFilterV1, 3); | |||||
| using LocalV2 = Local; | |||||
| using LocalBackwardDataV2 = LocalBackwardData; | |||||
| using LocalBackwardFilterV2 = LocalBackwardFilter; | |||||
| MGB_SEREG_OPR(LocalV2, 2); | |||||
| MGB_SEREG_OPR(LocalBackwardDataV2, 3); | |||||
| MGB_SEREG_OPR(LocalBackwardFilterV2, 3); | |||||
| using GroupLocalV2 = GroupLocal; | |||||
| using GroupLocalBackwardDataV2 = GroupLocalBackwardData; | |||||
| using GroupLocalBackwardFilterV2 = GroupLocalBackwardFilter; | |||||
| MGB_SEREG_OPR(GroupLocalV2, 2); | |||||
| MGB_SEREG_OPR(GroupLocalBackwardDataV2, 3); | |||||
| MGB_SEREG_OPR(GroupLocalBackwardFilterV2, 3); | |||||
| MGB_SEREG_OPR(LRN, 1); | MGB_SEREG_OPR(LRN, 1); | ||||
| MGB_SEREG_OPR(LRNBackward, 3); | MGB_SEREG_OPR(LRNBackward, 3); | ||||
| @@ -542,8 +542,8 @@ MGB_SEREG_OPR(AdaptivePoolingBackwardV1, 4); | |||||
| MGB_SEREG_OPR(ROIPooling, 3); | MGB_SEREG_OPR(ROIPooling, 3); | ||||
| MGB_SEREG_OPR(ROIPoolingBackward, 4); | MGB_SEREG_OPR(ROIPoolingBackward, 4); | ||||
| using MaskConvolutionV1 = MaskConvolution; | |||||
| MGB_SEREG_OPR(MaskConvolutionV1, 3); | |||||
| using MaskConvolutionV2 = MaskConvolution; | |||||
| MGB_SEREG_OPR(MaskConvolutionV2, 3); | |||||
| MGB_SEREG_OPR(MaskPropagate, 1); | MGB_SEREG_OPR(MaskPropagate, 1); | ||||
| MGB_SEREG_OPR(Convolution3D, 0); | MGB_SEREG_OPR(Convolution3D, 0); | ||||
| @@ -567,9 +567,12 @@ using ROIAlignV1 = ROIAlign; | |||||
| using ROIAlignBackwardV1 = ROIAlignBackward; | using ROIAlignBackwardV1 = ROIAlignBackward; | ||||
| MGB_SEREG_OPR(ROIAlignV1, 2); | MGB_SEREG_OPR(ROIAlignV1, 2); | ||||
| MGB_SEREG_OPR(ROIAlignBackwardV1, 4); | MGB_SEREG_OPR(ROIAlignBackwardV1, 4); | ||||
| MGB_SEREG_OPR(DeformableConvForward, 0); | |||||
| MGB_SEREG_OPR(DeformableConvBackwardData, 0); | |||||
| MGB_SEREG_OPR(DeformableConvBackwardFilter, 0); | |||||
| using DeformableConvForwardV1 = DeformableConvForward; | |||||
| using DeformableConvBackwardDataV1 = DeformableConvBackwardData; | |||||
| using DeformableConvBackwardFilterV1 = DeformableConvBackwardFilter; | |||||
| MGB_SEREG_OPR(DeformableConvForwardV1, 0); | |||||
| MGB_SEREG_OPR(DeformableConvBackwardDataV1, 0); | |||||
| MGB_SEREG_OPR(DeformableConvBackwardFilterV1, 0); | |||||
| MGB_SEREG_OPR(DeformablePSROIPoolingForward, 3); | MGB_SEREG_OPR(DeformablePSROIPoolingForward, 3); | ||||
| MGB_SEREG_OPR(DeformablePSROIPoolingBackward, 5); | MGB_SEREG_OPR(DeformablePSROIPoolingBackward, 5); | ||||
| @@ -24,8 +24,8 @@ decl_opr( | |||||
| params='WarpPerspective', | params='WarpPerspective', | ||||
| desc='like :func:`warp_perspective`, but the **mat** param does not have ' | desc='like :func:`warp_perspective`, but the **mat** param does not have ' | ||||
| 'to have the same batch size as **src**; instead, **mat_idx** specifies ' | 'to have the same batch size as **src**; instead, **mat_idx** specifies ' | ||||
| 'the batch number of each matrix in **mat**' | |||||
| ) | |||||
| 'the batch number of each matrix in **mat**', | |||||
| version=2) | |||||
| decl_opr('Rotate', | decl_opr('Rotate', | ||||
| inputs=[ | inputs=[ | ||||
| @@ -111,7 +111,7 @@ decl_raw_opr( | |||||
| 'assert not kwargs, "extra kwargs: {}".format(kwargs)', | 'assert not kwargs, "extra kwargs: {}".format(kwargs)', | ||||
| 'all_params = []', | 'all_params = []', | ||||
| 'all_params.append(param.serialize())', | 'all_params.append(param.serialize())', | ||||
| 'output = _mgb._create_opr("DctChannelSelect", all_inputs, all_params, config)', | |||||
| 'output = _mgb._create_opr("DctChannelSelectV1", all_inputs, all_params, config)', | |||||
| ], | ], | ||||
| has_out_dtype=True, | has_out_dtype=True, | ||||
| desc='DctChannelSelect do DCT with channel select' | desc='DctChannelSelect do DCT with channel select' | ||||