Searched refs:cell_state (Results 1 – 9 of 9) sorted by relevance
| /cmsis-nn-3.7.0/Source/NNSupportFunctions/ |
| D | arm_nn_lstm_step_s16.c | 59 int16_t *cell_state = buffers->cell_state; in arm_nn_lstm_step_s16() local
65 cell_state, in arm_nn_lstm_step_s16()
68 cell_state, in arm_nn_lstm_step_s16()
85 cell_state, in arm_nn_lstm_step_s16()
97 …cell_state, hidden_temp, params->hidden_size * params->batch_size, params->cell_scale_power + 12, … in arm_nn_lstm_step_s16()
|
| D | arm_nn_lstm_step_s8.c | 58 int16_t *cell_state = buffers->cell_state; in arm_nn_lstm_step_s8() local
64 cell_state, in arm_nn_lstm_step_s8()
67 cell_state, in arm_nn_lstm_step_s8()
83 cell_state, in arm_nn_lstm_step_s8()
95 …cell_state, hidden_temp, params->hidden_size * params->batch_size, params->cell_scale_power + 12, … in arm_nn_lstm_step_s8()
|
| /cmsis-nn-3.7.0/Source/LSTMFunctions/ |
| D | arm_lstm_unidirectional_s16.c | 56 memset(buffers->cell_state, 0, params->batch_size * params->hidden_size * sizeof(int16_t)); in arm_lstm_unidirectional_s16()
|
| D | arm_lstm_unidirectional_s8.c | 56 memset(buffers->cell_state, 0, params->batch_size * params->hidden_size * sizeof(int16_t)); in arm_lstm_unidirectional_s8()
|
| /cmsis-nn-3.7.0/Tests/UnitTest/TestCases/test_arm_lstm_unidirectional_s16/ |
| D | test_arm_lstm_unidirectional_s16.c | 175 buffers.cell_state = buffer3; in lstm_1_s16()
321 buffers.cell_state = buffer3; in lstm_2_s16()
469 buffers.cell_state = buffer3; in lstm_one_time_step_s16()
|
| /cmsis-nn-3.7.0/Tests/UnitTest/TestCases/test_arm_lstm_unidirectional_s8/ |
| D | test_arm_lstm_unidirectional_s8.c | 185 buffers.cell_state = buffer3; in lstm_1_arm_lstm_unidirectional_s8()
336 buffers.cell_state = buffer3; in lstm_2_arm_lstm_unidirectional_s8()
489 buffers.cell_state = buffer3; in lstm_one_time_step_arm_lstm_unidirectional_s8()
|
| /cmsis-nn-3.7.0/Include/ |
| D | arm_nn_types.h | 248 void *cell_state; member
|
| /cmsis-nn-3.7.0/Tests/UnitTest/RefactoredTestGen/Lib/ |
| D | op_lstm.py | 95 cell_state = tensor_details[14 + time_major_offset * 2]
96 scales["cell_scale"] = cell_state['quantization_parameters']['scales'][0]
|
| /cmsis-nn-3.7.0/Tests/UnitTest/ |
| D | lstm_settings.py | 202 cell_state = all_layers_details[self.cell_state_index + time_major_offset]
247 …self.generate_c_array("cell_state", interpreter.get_tensor(cell_state['index']), datatype='int16_t…
256 cell_scale = cell_state['quantization_parameters']['scales'][0]
|
