Searched refs:weights_feature_dims (Results 1 – 6 of 6) sorted by relevance
/cmsis-nn-latest/Source/SVDFunctions/ |
D | arm_svdf_get_buffer_sizes_s8.c | 42 int32_t arm_svdf_s8_get_buffer_size_dsp(const cmsis_nn_dims *weights_feature_dims) in arm_svdf_s8_get_buffer_size_dsp() argument 44 (void)weights_feature_dims; in arm_svdf_s8_get_buffer_size_dsp() 48 int32_t arm_svdf_s8_get_buffer_size_mve(const cmsis_nn_dims *weights_feature_dims) in arm_svdf_s8_get_buffer_size_mve() argument 50 return weights_feature_dims->n * sizeof(int32_t); in arm_svdf_s8_get_buffer_size_mve() 53 int32_t arm_svdf_s8_get_buffer_size(const cmsis_nn_dims *weights_feature_dims) in arm_svdf_s8_get_buffer_size() argument 56 return arm_svdf_s8_get_buffer_size_mve(weights_feature_dims); in arm_svdf_s8_get_buffer_size() 58 return arm_svdf_s8_get_buffer_size_dsp(weights_feature_dims); in arm_svdf_s8_get_buffer_size()
|
D | arm_svdf_state_s16_s8.c | 59 const cmsis_nn_dims *weights_feature_dims, in arm_svdf_state_s16_s8() argument 86 const int32_t feature_batches = weights_feature_dims->n; in arm_svdf_state_s16_s8()
|
D | arm_svdf_s8.c | 60 const cmsis_nn_dims *weights_feature_dims, in arm_svdf_s8() argument 94 const int32_t feature_batches = weights_feature_dims->n; in arm_svdf_s8()
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_svdf_s8/ |
D | test_arm_svdf_s8.c | 37 cmsis_nn_dims weights_feature_dims; in svdf_int8_arm_svdf_s8() local 50 weights_feature_dims.n = SVDF_INT8_FEATURE_BATCHES; in svdf_int8_arm_svdf_s8() 73 const int32_t buf_size = arm_svdf_s8_get_buffer_size(&weights_feature_dims); in svdf_int8_arm_svdf_s8() 79 …arm_vector_sum_s8(kernel_sum_buf, input_dims.h, weights_feature_dims.n, weights_feature_data, 1, N… in svdf_int8_arm_svdf_s8() 110 &weights_feature_dims, in svdf_int8_arm_svdf_s8() 152 cmsis_nn_dims weights_feature_dims; in svdf_int8_2_arm_svdf_s8() local 165 weights_feature_dims.n = SVDF_INT8_2_FEATURE_BATCHES; in svdf_int8_2_arm_svdf_s8() 188 const int32_t buf_size = arm_svdf_s8_get_buffer_size(&weights_feature_dims); in svdf_int8_2_arm_svdf_s8() 194 …arm_vector_sum_s8(kernel_sum_buf, input_dims.h, weights_feature_dims.n, weights_feature_data, 1, N… in svdf_int8_2_arm_svdf_s8() 221 &weights_feature_dims, in svdf_int8_2_arm_svdf_s8()
|
/cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_svdf_state_s16_s8/ |
D | test_arm_svdf_state_s16_s8.c | 51 cmsis_nn_dims weights_feature_dims; in svdf_arm_svdf_state_s16_s8() local 64 weights_feature_dims.n = SVDF_FEATURE_BATCHES; in svdf_arm_svdf_state_s16_s8() 111 &weights_feature_dims, in svdf_arm_svdf_state_s16_s8() 137 cmsis_nn_dims weights_feature_dims; in svdf_1_arm_svdf_state_s16_s8() local 150 weights_feature_dims.n = SVDF_1_FEATURE_BATCHES; in svdf_1_arm_svdf_state_s16_s8() 196 &weights_feature_dims, in svdf_1_arm_svdf_state_s16_s8() 222 cmsis_nn_dims weights_feature_dims; in svdf_2_arm_svdf_state_s16_s8() local 235 weights_feature_dims.n = SVDF_2_FEATURE_BATCHES; in svdf_2_arm_svdf_state_s16_s8() 281 &weights_feature_dims, in svdf_2_arm_svdf_state_s16_s8() 307 cmsis_nn_dims weights_feature_dims; in svdf_3_arm_svdf_state_s16_s8() local [all …]
|
/cmsis-nn-latest/Include/ |
D | arm_nnfunctions.h | 2433 const cmsis_nn_dims *weights_feature_dims, 2481 const cmsis_nn_dims *weights_feature_dims,
|