add reflection padding from github

2022-01-03 18:58:13 +05:30
parent dcf4054bc6
commit 6837644eb2
2 changed files with 72 additions and 0 deletions
@@ -48,4 +48,8 @@ void dcnV2CudaForward(cublasStatus_t stat, cublasHandle_t handle,
                         const int dst_dim, cudaStream_t stream = cudaStream_t(0));

 void scalAdd(dnnType* dstData, int size, float alpha, float beta, int inc, cudaStream_t stream = cudaStream_t(0));
+
+void reflection_pad2d_out_forward(int32_t padding[4], float* srcData, float* dstData, int32_t input_h, int32_t input_w, int32_t plane_dim, int32_t n_batch, cudaStream_t cudaStream = cudaStream_t(0));
+
+
 #endif //KERNELS_H