diff --git a/src/operators/gru_unit_op.cpp b/src/operators/gru_unit_op.cpp index 056026c5b9fb359de378ae2d63d13c0fac16e367..38ddd217943fd63ce457b62a8c695ba0fa684b8d 100644 --- a/src/operators/gru_unit_op.cpp +++ b/src/operators/gru_unit_op.cpp @@ -67,4 +67,4 @@ REGISTER_OPERATOR_CPU(gru_unit, ops::GruUnitOp); #ifdef PADDLE_MOBILE_CL #endif -#endif \ No newline at end of file +#endif diff --git a/src/operators/gru_unit_op.h b/src/operators/gru_unit_op.h index c9be667906b4800d0b91f3a9a92925be2ea3ea29..4188662d05e79a97fa2f0dba62303391ae8e0d70 100644 --- a/src/operators/gru_unit_op.h +++ b/src/operators/gru_unit_op.h @@ -40,4 +40,4 @@ class GruUnitOp : public framework::OperatorWithKernel< } // namespace operators } // namespace paddle_mobile -#endif \ No newline at end of file +#endif diff --git a/src/operators/kernel/central-arm-func/gru_unit_arm_func.h b/src/operators/kernel/central-arm-func/gru_unit_arm_func.h index 599b9b46ceb50c3d3d4bd3e9be666823882efb5b..af97ef750723996084a8130cc180e829a39a4aac 100644 --- a/src/operators/kernel/central-arm-func/gru_unit_arm_func.h +++ b/src/operators/kernel/central-arm-func/gru_unit_arm_func.h @@ -47,7 +47,7 @@ void GruUnitCompute(const GruUnitParam& param) { gru_value.state_weight = const_cast(weight_data + 2 * frame_size * frame_size); gru_value.output_value = hidden->data

(); - gru_value.prev_out_value = gru_value.output_value; + gru_value.prev_out_value = hidden_prev->data

(); gru_value.gate_value = gate->data

(); gru_value.reset_output_value = reset_hidden_prev->data

(); auto active_node = math::GetActivationType(param.Activation()); @@ -59,4 +59,4 @@ void GruUnitCompute(const GruUnitParam& param) { } // namespace operators } // namespace paddle_mobile -#endif \ No newline at end of file +#endif diff --git a/src/operators/kernel/gru_unit_kernel.h b/src/operators/kernel/gru_unit_kernel.h index 53dfa234a79e4fe53e22c034c509ddc59130ce98..bda17cd20504d057e4a824e6ac046cf35d240d9d 100644 --- a/src/operators/kernel/gru_unit_kernel.h +++ b/src/operators/kernel/gru_unit_kernel.h @@ -32,4 +32,4 @@ class GruUnitKernel } // namespace operators } // namespace paddle_mobile -#endif \ No newline at end of file +#endif