|
|
|
@@ -70,7 +70,7 @@ void PostConvFuncFp32C4(const float *c4_out_ptr, float *out_ptr, const float *bi |
|
|
|
|
|
|
|
#ifndef ENABLE_ARM |
|
|
|
void WinogradTransLeft(const float *S, const float *B, float *M, size_t w, size_t h, size_t k, size_t length) { |
|
|
|
int unitStep = 4 * length; |
|
|
|
const int unitStep = 4 * length; |
|
|
|
for (int y = 0; y < h; ++y) { |
|
|
|
float *dstY = M + y * w * unitStep; |
|
|
|
for (int x = 0; x < w; ++x) { |
|
|
|
@@ -93,7 +93,7 @@ void WinogradTransLeft(const float *S, const float *B, float *M, size_t w, size_ |
|
|
|
|
|
|
|
// M = S * B , M = w*h * l, S = k*h * l, B = w*k |
|
|
|
void WinogradTransRight(const float *S, const float *B, float *M, size_t w, size_t h, size_t k, size_t length) { |
|
|
|
int unitStep = 4 * length; |
|
|
|
const int unitStep = 4 * length; |
|
|
|
for (int y = 0; y < h; ++y) { |
|
|
|
float *dstY = M + y * w * unitStep; |
|
|
|
const float *srcY = S + y * k * unitStep; |
|
|
|
|