Skip to content

Commit 0e6b303

Browse files
committed
MKLDNN Pooling: inline functions handling ceiled mode
1 parent f465b03 commit 0e6b303

File tree

1 file changed

+8
-9
lines changed

1 file changed

+8
-9
lines changed

paddle/fluid/operators/pool_mkldnn_op.cc

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -46,18 +46,17 @@ static std::string gethash(const memory::dims& input_dims,
4646
dims2str(paddings) + pooling_type + suffix;
4747
}
4848

49-
static int ComputeCeiledOutput(int input_size, int kernel_size, int padding,
50-
int stride) {
49+
static inline int ComputeCeiledOutput(int input_size, int kernel_size,
50+
int padding, int stride) {
5151
return (input_size - kernel_size + 2 * padding) / stride + 1;
5252
}
5353

54-
static void CorrectOutputSize(const std::vector<int>& src_tz,
55-
const std::vector<int>& dst_tz,
56-
const std::vector<int>& kernel_size,
57-
const std::vector<int>& paddings,
58-
const std::vector<int>& strides,
59-
std::vector<int>& right_bot_padding) {
60-
for (int i = 0; i < right_bot_padding.size(); i++) {
54+
static inline void CorrectOutputSize(
55+
const std::vector<int>& src_tz, const std::vector<int>& dst_tz,
56+
const std::vector<int>& kernel_size, const std::vector<int>& paddings,
57+
const std::vector<int>& strides,
58+
std::vector<int>& right_bot_padding) { // NOLINT
59+
for (size_t i = 0; i < right_bot_padding.size(); i++) {
6160
int desired_size = ComputeCeiledOutput(src_tz[i + 2], kernel_size[i],
6261
paddings[i], strides[i]);
6362
if (desired_size != dst_tz[i + 2]) {

0 commit comments

Comments
 (0)