Searched refs:hidden_weights (Results 1 – 5 of 5) sorted by relevance
| /cmsis-nn-latest/Source/NNSupportFunctions/ |
| D | arm_nn_lstm_calculate_gate_s16.c | 72 gate->hidden_weights, in arm_nn_lstm_calculate_gate_s16()
|
| D | arm_nn_lstm_calculate_gate_s8_s16.c | 78 gate->hidden_weights, in arm_nn_lstm_calculate_gate_s8_s16()
|
| /cmsis-nn-latest/Tests/UnitTest/ |
| D | lstm_settings.py | 97 … def generate_data(self, input_data=None, weights=None, hidden_weights=None, biases=None) -> None: argument
123 if hidden_weights is not None:
124 hidden_weights = tf.reshape(hidden_weights, [number_cells, number_cells * number_w_b])
126 hidden_weights = self.get_randomized_data([number_cells, number_cells * number_w_b],
170 model.layers[1 + time_major_offset].weights[1].assign(hidden_weights)
|
| /cmsis-nn-latest/Include/ |
| D | arm_nn_types.h | 207 const void *hidden_weights; member
|
| /cmsis-nn-latest/Tests/UnitTest/RefactoredTestGen/Lib/ |
| D | op_lstm.py | 70 … hidden_weights = Lib.op_utils.generate_tf_tensor(shapes["all_hidden_weights"], -1, 1, decimals=8)
71 model.layers[1 + time_major_offset].weights[1].assign(hidden_weights)
|
