未验证 提交 5fcb6f63 编写于 作者: M mapingshuo 提交者: GitHub

fix sequence_conv bug for multi batch samples (#3824)

上级 bff64fb1
...@@ -88,7 +88,7 @@ void SequenceConvCompute::Run() { ...@@ -88,7 +88,7 @@ void SequenceConvCompute::Run() {
paddle::lite::arm::math::im2col( paddle::lite::arm::math::im2col(
sub_in_data, sub_in_data,
1, 1,
sequence_len, input_row_end - input_row_begin,
hidden_dim, // C H W -> 1, seq_len, hidden_dim hidden_dim, // C H W -> 1, seq_len, hidden_dim
kernel_size, kernel_size,
hidden_dim, // kernel_h, kernel_w hidden_dim, // kernel_h, kernel_w
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册