Skip to content

P
Paddle

BaiXuePrincess / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 0e6b303f 编写于 作者: M Michal Gallus
浏览文件
操作

MKLDNN Pooling: inline functions handling ceiled mode

上级 f465b03e
隐藏空白更改
内联 并排
Showing with 8 addition and 9 deletion
paddle/fluid/operators/pool_mkldnn_op.cc
浏览文件 @ 0e6b303f
...@@ -46,18 +46,17 @@ static std::string gethash(const memory::dims& input_dims, ...@@ -46,18 +46,17 @@ static std::string gethash(const memory::dims& input_dims,
dims2str(paddings) + pooling_type + suffix; dims2str(paddings) + pooling_type + suffix;
} }
static int ComputeCeiledOutput(int input_size, int kernel_size, int padding, static inline int ComputeCeiledOutput(int input_size, int kernel_size,
int stride) { int padding, int stride) {
return (input_size - kernel_size + 2 * padding) / stride + 1; return (input_size - kernel_size + 2 * padding) / stride + 1;
} }
static void CorrectOutputSize(const std::vector<int>& src_tz, static inline void CorrectOutputSize(
const std::vector<int>& dst_tz, const std::vector<int>& src_tz, const std::vector<int>& dst_tz,
const std::vector<int>& kernel_size, const std::vector<int>& kernel_size, const std::vector<int>& paddings,
const std::vector<int>& paddings, const std::vector<int>& strides,
const std::vector<int>& strides, std::vector<int>& right_bot_padding) { // NOLINT
std::vector<int>& right_bot_padding) { for (size_t i = 0; i < right_bot_padding.size(); i++) {
for (int i = 0; i < right_bot_padding.size(); i++) {
int desired_size = ComputeCeiledOutput(src_tz[i + 2], kernel_size[i], int desired_size = ComputeCeiledOutput(src_tz[i + 2], kernel_size[i],
paddings[i], strides[i]); paddings[i], strides[i]);
if (desired_size != dst_tz[i + 2]) { if (desired_size != dst_tz[i + 2]) {
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或