Home
last modified time | relevance | path

Searched refs:feature_batches (Results 1 – 2 of 2) sorted by relevance

/cmsis-nn-3.6.0-3.5.0/Source/SVDFunctions/
Darm_svdf_state_s16_s8.c86 const int32_t feature_batches = weights_feature_dims->n; in arm_svdf_state_s16_s8() local
88 const int32_t unit_count = feature_batches / rank; in arm_svdf_state_s16_s8()
105 … (size_t)((input_batches * feature_batches * time_batches - 1) * (int32_t)sizeof(int16_t))); in arm_svdf_state_s16_s8()
110 … int16_t *res_ptr = state_data + (time_batches * i_batch * feature_batches) + (time_batches - 1); in arm_svdf_state_s16_s8()
122 feature_batches, in arm_svdf_state_s16_s8()
140 for (int i_feature_batch = 0; i_feature_batch < feature_batches; i_feature_batch++) in arm_svdf_state_s16_s8()
181 if (unit_count == feature_batches) in arm_svdf_state_s16_s8()
185 int32_t *output_temp = buffer_b + i * feature_batches; in arm_svdf_state_s16_s8()
186 const int32_t *ptr_a = buffer_a + i * feature_batches; in arm_svdf_state_s16_s8()
189 for (int j = 0; j < feature_batches; j++) in arm_svdf_state_s16_s8()
[all …]
Darm_svdf_s8.c86 const int32_t feature_batches = weights_feature_dims->n; in arm_svdf_s8() local
88 const int32_t unit_count = feature_batches / rank; in arm_svdf_s8()
105 … (size_t)((input_batches * feature_batches * time_batches - 1) * (int32_t)sizeof(int8_t))); in arm_svdf_s8()
110 … int8_t *res_ptr = state_data + (time_batches * i_batch * feature_batches) + (time_batches - 1); in arm_svdf_s8()
123 feature_batches, in arm_svdf_s8()
142 for (int i_feature_batch = 0; i_feature_batch < feature_batches; i_feature_batch++) in arm_svdf_s8()
185 if (unit_count == feature_batches) in arm_svdf_s8()
189 int32_t *output_temp = buffer_b + i * feature_batches; in arm_svdf_s8()
190 const int32_t *ptr_a = buffer_a + i * feature_batches; in arm_svdf_s8()
193 for (int j = 0; j < feature_batches; j++) in arm_svdf_s8()
[all …]