Searched refs:forget_gate (Results 1 – 3 of 3) sorted by relevance
| /cmsis-nn-latest/Source/NNSupportFunctions/ |
| D | arm_nn_lstm_step_s16.c | 53 int16_t *forget_gate = buffers->temp1; in arm_nn_lstm_step_s16() local
61 …arm_nn_lstm_calculate_gate_s16(data_in, hidden_in, ¶ms->forget_gate, params, forget_gate, batc… in arm_nn_lstm_step_s16()
64 arm_elementwise_mul_s16(forget_gate, in arm_nn_lstm_step_s16()
81 arm_elementwise_mul_acc_s16(forget_gate, in arm_nn_lstm_step_s16()
|
| D | arm_nn_lstm_step_s8.c | 52 int16_t *forget_gate = buffers->temp1; in arm_nn_lstm_step_s8() local
60 …arm_nn_lstm_calculate_gate_s8_s16(data_in, hidden_in, ¶ms->forget_gate, params, forget_gate, b… in arm_nn_lstm_step_s8()
63 arm_elementwise_mul_s16(forget_gate, in arm_nn_lstm_step_s8()
79 arm_elementwise_mul_acc_s16(forget_gate, in arm_nn_lstm_step_s8()
|
| /cmsis-nn-latest/Include/ |
| D | arm_nn_types.h | 237 cmsis_nn_lstm_gate forget_gate; member
|
