|
|
|
@ -87,7 +87,8 @@ inline void UpdatePadding(std::vector<T>* paddings, const bool global_pooling,
|
|
|
|
|
for (int i = 0; i < data_dims.size(); ++i) {
|
|
|
|
|
T out_size = (data_dims[i] + strides[i] - 1) / strides[i];
|
|
|
|
|
T pad_sum =
|
|
|
|
|
std::max((out_size - 1) * strides[i] + ksize[i] - data_shape[i], 0);
|
|
|
|
|
std::max((out_size - 1) * strides[i] + ksize[i] - data_shape[i],
|
|
|
|
|
static_cast<T>(0));
|
|
|
|
|
T pad_0 = pad_sum / 2;
|
|
|
|
|
T pad_1 = pad_sum - pad_0;
|
|
|
|
|
*(paddings->begin() + i * 2) = pad_0;
|
|
|
|
|