diff --git a/src/layer/arm/rnn_arm.cpp b/src/layer/arm/rnn_arm.cpp index 64294a0c274..895f16bb913 100644 --- a/src/layer/arm/rnn_arm.cpp +++ b/src/layer/arm/rnn_arm.cpp @@ -332,6 +332,8 @@ static int rnn_int8(const Mat& bottom_blob, Mat& top_blob, int reverse, const Ma int num_output = top_blob.w; + NCNN_LOGE("rnn_int8 %d %d %d", size, T, num_output); + // num_output Mat gates(num_output, 4u, opt.workspace_allocator); if (gates.empty()) @@ -1049,6 +1051,8 @@ static int rnn_bf16s_int8(const Mat& bottom_blob, Mat& top_blob, int reverse, co int num_output = top_blob.w; + NCNN_LOGE("rnn_bf16s_int8 %d %d %d", size, T, num_output); + // num_output Mat gates(num_output, 4u, opt.workspace_allocator); if (gates.empty())