提交 ca3c0309 编写于 作者: Z ZhenWang

fix some bugs.

上级 9a30bf0b
...@@ -34,12 +34,12 @@ struct GRUUnitFunctor<CPU, T> { ...@@ -34,12 +34,12 @@ struct GRUUnitFunctor<CPU, T> {
gemm.Sgemm_omp(batch_size, frame_size * 2, frame_size, 1, gemm.Sgemm_omp(batch_size, frame_size * 2, frame_size, 1,
value.prev_out_value, frame_size, value.gate_weight, value.prev_out_value, frame_size, value.gate_weight,
frame_size * 2, 1, value.gate_value, frame_size * 3, false, frame_size * 2, 1, value.gate_value, frame_size * 3, false,
nullptr); static_cast<float *>(nullptr));
#else #else
gemm.Sgemm(batch_size, frame_size * 2, frame_size, 1, gemm.Sgemm(batch_size, frame_size * 2, frame_size, 1,
value.prev_out_value, frame_size, value.gate_weight, value.prev_out_value, frame_size, value.gate_weight,
frame_size * 2, 1, value.gate_value, frame_size * 3, false, frame_size * 2, 1, value.gate_value, frame_size * 3, false,
nullptr); static_cast<float *>(nullptr));
#endif #endif
} }
...@@ -51,12 +51,12 @@ struct GRUUnitFunctor<CPU, T> { ...@@ -51,12 +51,12 @@ struct GRUUnitFunctor<CPU, T> {
gemm.Sgemm_omp(batch_size, frame_size, frame_size, 1, gemm.Sgemm_omp(batch_size, frame_size, frame_size, 1,
value.reset_output_value, frame_size, value.state_weight, value.reset_output_value, frame_size, value.state_weight,
frame_size, 1, value.gate_value + frame_size * 2, frame_size, 1, value.gate_value + frame_size * 2,
frame_size * 3, false, nullptr); frame_size * 3, false, static_cast<float *>(nullptr));
#else #else
gemm.Sgemm(batch_size, frame_size, frame_size, 1, gemm.Sgemm(batch_size, frame_size, frame_size, 1,
value.reset_output_value, frame_size, value.state_weight, value.reset_output_value, frame_size, value.state_weight,
frame_size, 1, value.gate_value + frame_size * 2, frame_size, 1, value.gate_value + frame_size * 2,
frame_size * 3, false, nullptr); frame_size * 3, false, static_cast<float *>(nullptr));
#endif #endif
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册