提交 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,
};
} else if (use_neon_1x1_s1) {
conv_func = [=](const float *pad_input, float *pad_output) {
Conv2dNeonK1x1S1(input_data,
Conv2dNeonK1x1S1(pad_input,
filter_data,
batch,
height,
width,
extra_input_height,
extra_input_width,
input_channels,
channels,
output_data);
pad_output);
};
} else {
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.
先完成此消息的编辑!
想要评论请 注册