提交 4b4e17a6 编写于 作者: 李寅

Merge branch 'fix_1x1' into 'master'

Fix conv_2d 1x1 test breakage

See merge request !62
......@@ -38,7 +38,7 @@ void Conv2dNeonK1x1S1(const float *input, // NCHW
#pragma omp parallel for collapse(2)
for (index_t n = 0; n < batch; ++n) {
for (index_t c = 0; c < channels; ++c) {
const float *filter_ptr = filter;
const float *filter_ptr = filter + c * input_channels;
// TODO Will GCC opt these out?
float *channel_output_start =
output + n * channels * height * width + c * height * width;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册