From 5fcb6f6317f2241f914bc5d53cec3f05098eb1e8 Mon Sep 17 00:00:00 2001 From: mapingshuo Date: Mon, 22 Jun 2020 19:43:39 +0800 Subject: [PATCH] fix sequence_conv bug for multi batch samples (#3824) --- lite/kernels/arm/sequence_conv_compute.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lite/kernels/arm/sequence_conv_compute.cc b/lite/kernels/arm/sequence_conv_compute.cc index a70b671709..69740a258b 100644 --- a/lite/kernels/arm/sequence_conv_compute.cc +++ b/lite/kernels/arm/sequence_conv_compute.cc @@ -88,7 +88,7 @@ void SequenceConvCompute::Run() { paddle::lite::arm::math::im2col( sub_in_data, 1, - sequence_len, + input_row_end - input_row_begin, hidden_dim, // C H W -> 1, seq_len, hidden_dim kernel_size, hidden_dim, // kernel_h, kernel_w -- GitLab