Searched refs:number_units (Results 1 – 7 of 7) sorted by relevance
| /cmsis-nn-latest/Tests/UnitTest/TestCases/Common/ |
| D | svdf_s8_weights_template.json | 71 number_units
109 number_units
|
| D | svdf_template.json | 71 number_units
109 number_units
|
| /cmsis-nn-latest/Tests/UnitTest/ |
| D | svdf_settings.py | 38 number_units=4, argument
74 self.number_units = number_units
78 self.number_filters = self.number_units * self.rank
102 "number_units": self.number_units,
142 f.write("#define {}_DST_SIZE {}\n".format(prefix, self.number_units * self.batches))
183 biases = [0] * self.number_units
185 biases = tf.reshape(biases, [self.number_units])
187 biases = self.get_randomized_data([self.number_units],
|
| D | lstm_settings.py | 36 number_units=4, argument
64 self.number_units = number_units
108 number_cells = self.number_units
150 lstm_layer = keras.layers.LSTM(units=self.number_units,
154 lstm_layer = keras.layers.LSTM(units=self.number_units,
371 f.write("#define {}_BUFFER_SIZE {}\n".format(prefix, self.batches * self.number_units))
373 …rite("#define {}_DST_SIZE {}\n".format(prefix, self.batches * self.time_steps * self.number_units))
375 f.write("#define {}_NUMBER_UNITS {}\n".format(prefix, self.number_units))
|
| D | generate_test_data.py | 2999 number_units=3,
3013 number_units=5,
3027 number_units=5,
3042 number_units=12,
3057 number_units=12,
3073 number_units=13,
3183 number_units=11,
3196 number_units=7,
3209 number_units=3,
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_svdf_state_s16_s8/ |
| D | test_arm_svdf_state_s16_s8.c | 84 const int32_t number_units = SVDF_FEATURE_BATCHES / SVDF_RANK; in svdf_arm_svdf_state_s16_s8() local
86 const int scratch_size_out = SVDF_INPUT_BATCHES * number_units * sizeof(int32_t); in svdf_arm_svdf_state_s16_s8()
170 const int32_t number_units = SVDF_1_FEATURE_BATCHES / SVDF_1_RANK; in svdf_1_arm_svdf_state_s16_s8() local
172 const int scratch_size_out = SVDF_1_INPUT_BATCHES * number_units * sizeof(int32_t); in svdf_1_arm_svdf_state_s16_s8()
255 const int32_t number_units = SVDF_2_FEATURE_BATCHES / SVDF_2_RANK; in svdf_2_arm_svdf_state_s16_s8() local
257 const int scratch_size_out = SVDF_2_INPUT_BATCHES * number_units * sizeof(int32_t); in svdf_2_arm_svdf_state_s16_s8()
340 const int32_t number_units = SVDF_3_FEATURE_BATCHES / SVDF_3_RANK; in svdf_3_arm_svdf_state_s16_s8() local
342 const int scratch_size_out = SVDF_3_INPUT_BATCHES * number_units * sizeof(int32_t); in svdf_3_arm_svdf_state_s16_s8()
|
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_svdf_s8/ |
| D | test_arm_svdf_s8.c | 68 const int32_t number_units = SVDF_INT8_FEATURE_BATCHES / SVDF_INT8_RANK; in svdf_int8_arm_svdf_s8() local
70 const int scratch_size_out = SVDF_INT8_INPUT_BATCHES * number_units * sizeof(int32_t); in svdf_int8_arm_svdf_s8()
183 const int32_t number_units = SVDF_INT8_2_FEATURE_BATCHES / SVDF_INT8_2_RANK; in svdf_int8_2_arm_svdf_s8() local
185 const int scratch_size_out = SVDF_INT8_2_INPUT_BATCHES * number_units * sizeof(int32_t); in svdf_int8_2_arm_svdf_s8()
|
