提交 fed98beb 编写于 作者: L Liangliang He

Fix conv_2d 1x1 test breakage

上级 6b13f55a
...@@ -38,7 +38,7 @@ void Conv2dNeonK1x1S1(const float *input, // NCHW ...@@ -38,7 +38,7 @@ void Conv2dNeonK1x1S1(const float *input, // NCHW
#pragma omp parallel for collapse(2) #pragma omp parallel for collapse(2)
for (index_t n = 0; n < batch; ++n) { for (index_t n = 0; n < batch; ++n) {
for (index_t c = 0; c < channels; ++c) { 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? // TODO Will GCC opt these out?
float *channel_output_start = float *channel_output_start =
output + n * channels * height * width + c * height * width; 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.
先完成此消息的编辑!
想要评论请 注册