未验证 提交 832ed81a 编写于 作者: M mapingshuo 提交者: GitHub

fix sequence_conv bug for multi batch samples, test=develop (#3811)

上级 21bd3640
...@@ -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.
先完成此消息的编辑!
想要评论请 注册