@@ -217,10 +217,10 @@ void PrepareLSTMWeight(const LoDTensor& W_forget_w0,
217
217
float * out_data = out->mutable_data <float >(platform::CPUPlace ());
218
218
std::array<const float *, 4 > tensors (
219
219
{{W_forget_w0.data <float >(), W_input_w0.data <float >(),
220
- W_output_w0.data <float >(), W_cell_w0.data <float >()}});
220
+ W_output_w0.data <float >(), W_cell_w0.data <float >()}});
221
221
std::array<const float *, 4 > tensors1 (
222
222
{{W_forget_w1.data <float >(), W_input_w1.data <float >(),
223
- W_output_w1.data <float >(), W_cell_w1.data <float >()}});
223
+ W_output_w1.data <float >(), W_cell_w1.data <float >()}});
224
224
225
225
for (int row = 0 ; row < D; row++) {
226
226
for (int col = 0 ; col < 4 ; col++) {
@@ -244,7 +244,7 @@ void PrepareLSTMBias(const LoDTensor& B_forget, const LoDTensor& B_input,
244
244
LoDTensor* out) {
245
245
std::array<const float *, 4 > tensors (
246
246
{{B_forget.data <float >(), B_input.data <float >(), B_output.data <float >(),
247
- B_cell.data <float >()}});
247
+ B_cell.data <float >()}});
248
248
249
249
PADDLE_ENFORCE_EQ (B_forget.dims ().size (), 1 );
250
250
int D = B_forget.dims ()[0 ];
0 commit comments