File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -81,7 +81,7 @@ void RnnKernel(const Context& dev_ctx,
8181 int in_dim_arr[in_out_dim_num] = {seq_len, batch_size, input_dim};
8282 int out_dim_arr[in_out_dim_num] = {
8383 seq_len, batch_size, direction_num * hidden_size};
84- int proj_size = hidden_size ;
84+ int proj_size = 0 ;
8585
8686 std::vector<int > seq_len_vec (batch_size, seq_len);
8787 if (sequence_length.is_initialized ()) { // set seq_len if no padding,
@@ -380,7 +380,7 @@ void RnnGradKernel(const Context& dev_ctx,
380380 int in_dim_arr[in_out_dim_num] = {seq_len, batch_size, input_dim};
381381 int out_dim_arr[in_out_dim_num] = {
382382 seq_len, batch_size, direction_num * hidden_size};
383- int proj_size = hidden_size ;
383+ int proj_size = 0 ;
384384 PADDLE_ENFORCE_EQ (
385385 num_layers,
386386 1 ,
You can’t perform that action at this time.
0 commit comments