diff --git a/paddle/operators/math/im2col.h b/paddle/operators/math/im2col.h index c736d4fa523c2af3e3dd7a11114d7f84021bc5c1..a1cb956c51ebfa80892745b88a6d85a0f09202d6 100644 --- a/paddle/operators/math/im2col.h +++ b/paddle/operators/math/im2col.h @@ -75,7 +75,8 @@ class Im2ColFunctor { void operator()(const platform::DeviceContext& context, const framework::Tensor& im, framework::Tensor& col, int stride_height, int stride_width, int padding_up, - int padding_down, int padding_left, int padding_right); + int padding_down, int padding_left = 0, + int padding_right = 0); }; template @@ -84,7 +85,7 @@ class Col2ImFunctor { void operator()(const platform::DeviceContext& context, framework::Tensor& im, const framework::Tensor& col, int stride_height, int stride_width, int padding_up, int padding_down, - int padding_left, int padding_right); + int padding_left = 0, int padding_right = 0); }; } // namespace math