97 void compute_gru(const GRULayer *gru, float *state, const float *input)  in compute_gru()  argument
105    M = gru->nb_inputs;  in compute_gru()
106    N = gru->nb_neurons;  in compute_gru()
111       float sum = gru->bias[i];  in compute_gru()
113          sum += gru->input_weights[j*stride + i]*input[j];  in compute_gru()
115          sum += gru->recurrent_weights[j*stride + i]*state[j];  in compute_gru()
121       float sum = gru->bias[N + i];  in compute_gru()
123          sum += gru->input_weights[N + j*stride + i]*input[j];  in compute_gru()
125          sum += gru->recurrent_weights[N + j*stride + i]*state[j];  in compute_gru()
131       float sum = gru->bias[2*N + i];  in compute_gru()
133          sum += gru->input_weights[2*N + j*stride + i]*input[j];  in compute_gru()
135          sum += gru->recurrent_weights[2*N + j*stride + i]*state[j]*r[j];  in compute_gru()