deformable conv cu

2019-10-30 10:20:04 +01:00
parent 02936fa928
commit 8b64d1876f
3 changed files with 148 additions and 0 deletions
@@ -24,4 +24,13 @@ void upsampleForward(dnnType* srcData, dnnType* dstData,
                     cudaStream_t stream = cudaStream_t(0));

 void float2half(float* srcData, __half* dstData, int size, const cudaStream_t stream = cudaStream_t(0));
+
+
+void modulated_deformable_im2col_cuda(cudaStream_t stream,
+                                const float *data_im, const float *data_offset, const float *data_mask,
+                                const int batch_size, const int channels, const int height_im, const int width_im,
+                                const int height_col, const int width_col, const int kernel_h, const int kenerl_w,
+                                const int pad_h, const int pad_w, const int stride_h, const int stride_w,
+                                const int dilation_h, const int dilation_w,
+                                const int deformable_group, float *data_col);
 #endif //KERNELS_H