提交 9f867daf 编写于 作者: X xiebaiyuan 提交者: GitHub

support global pooling ... test=develop (#2204)

上级 17833acb
...@@ -50,6 +50,14 @@ void PoolKernel<GPU_CL, float>::Compute(const PoolParam<GPU_CL> &param) { ...@@ -50,6 +50,14 @@ void PoolKernel<GPU_CL, float>::Compute(const PoolParam<GPU_CL> &param) {
std::vector<int> ksize = param.Ksize(); std::vector<int> ksize = param.Ksize();
std::vector<int> strides = param.Strides(); std::vector<int> strides = param.Strides();
std::vector<int> paddings = param.Paddings(); std::vector<int> paddings = param.Paddings();
if (param.isGlobalPooling()) {
for (size_t i = 0; i < ksize.size(); ++i) {
paddings[i] = 0;
ksize[i] = static_cast<int>(param.Input()->dims()[i + 2]);
}
}
const int pad_top = paddings[0]; const int pad_top = paddings[0];
const int pad_left = paddings[1]; const int pad_left = paddings[1];
const int stride_h = strides[0]; const int stride_h = strides[0];
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册