Searched refs:output_gate (Results 1 – 3 of 3) sorted by relevance
| /cmsis-nn-latest/Source/NNSupportFunctions/ |
| D | arm_nn_lstm_step_s16.c | 56 int16_t *output_gate = buffers->temp1; in arm_nn_lstm_step_s16() local
93 …arm_nn_lstm_calculate_gate_s16(data_in, hidden_in, ¶ms->output_gate, params, output_gate, batc… in arm_nn_lstm_step_s16()
98 arm_elementwise_mul_s16_batch_offset(output_gate, in arm_nn_lstm_step_s16()
|
| D | arm_nn_lstm_step_s8.c | 55 int16_t *output_gate = buffers->temp1; in arm_nn_lstm_step_s8() local
91 …arm_nn_lstm_calculate_gate_s8_s16(data_in, hidden_in, ¶ms->output_gate, params, output_gate, b… in arm_nn_lstm_step_s8()
96 arm_elementwise_mul_s16_s8(output_gate, in arm_nn_lstm_step_s8()
|
| /cmsis-nn-latest/Include/ |
| D | arm_nn_types.h | 240 cmsis_nn_lstm_gate output_gate; member
|
