提交 d9dba05d 编写于 作者: 吴承辉

Merge branch 'master' into 'master'

Fix conv1x1 pad

See merge request !399
...@@ -362,14 +362,14 @@ void Conv2dFunctor<DeviceType::NEON, float>::operator()(const Tensor *input, ...@@ -362,14 +362,14 @@ void Conv2dFunctor<DeviceType::NEON, float>::operator()(const Tensor *input,
}; };
} else if (use_neon_1x1_s1) { } else if (use_neon_1x1_s1) {
conv_func = [=](const float *pad_input, float *pad_output) { conv_func = [=](const float *pad_input, float *pad_output) {
Conv2dNeonK1x1S1(input_data, Conv2dNeonK1x1S1(pad_input,
filter_data, filter_data,
batch, batch,
height, extra_input_height,
width, extra_input_width,
input_channels, input_channels,
channels, channels,
output_data); pad_output);
}; };
} else { } else {
conv_func = [=](const float *pad_input, float *pad_output) { conv_func = [=](const float *pad_input, float *pad_output) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册