@@ -46,18 +46,17 @@ static std::string gethash(const memory::dims& input_dims,
46
46
dims2str (paddings) + pooling_type + suffix;
47
47
}
48
48
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) {
51
51
return (input_size - kernel_size + 2 * padding) / stride + 1 ;
52
52
}
53
53
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++) {
61
60
int desired_size = ComputeCeiledOutput (src_tz[i + 2 ], kernel_size[i],
62
61
paddings[i], strides[i]);
63
62
if (desired_size != dst_tz[i + 2 ]) {
0 commit comments