提交 a7776510 编写于 作者: Z zhaojiaying01

fix gru_unit op

上级 c68b22a9
...@@ -47,7 +47,7 @@ void GruUnitCompute(const GruUnitParam<CPU>& param) { ...@@ -47,7 +47,7 @@ void GruUnitCompute(const GruUnitParam<CPU>& param) {
gru_value.state_weight = gru_value.state_weight =
const_cast<P*>(weight_data + 2 * frame_size * frame_size); const_cast<P*>(weight_data + 2 * frame_size * frame_size);
gru_value.output_value = hidden->data<P>(); gru_value.output_value = hidden->data<P>();
gru_value.prev_out_value = gru_value.output_value; gru_value.prev_out_value = hidden_prev->data<P>();
gru_value.gate_value = gate->data<P>(); gru_value.gate_value = gate->data<P>();
gru_value.reset_output_value = reset_hidden_prev->data<P>(); gru_value.reset_output_value = reset_hidden_prev->data<P>();
auto active_node = math::GetActivationType(param.Activation()); auto active_node = math::GetActivationType(param.Activation());
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册