Searched refs:transpose_params (Results 1 – 4 of 4) sorted by relevance
| /cmsis-nn-latest/Tests/UnitTest/TestCases/test_arm_transpose_s8/ |
| D | test_arm_transpose_s8.c | 50 const cmsis_nn_transpose_params transpose_params = {TRANSPOSE_DEFAULT_PERM_SIZE, perm}; in transpose_default_arm_transpose_s8() local
52 …us result = arm_transpose_s8(input_data, output_ptr, &input_dims, &output_dims, &transpose_params); in transpose_default_arm_transpose_s8()
72 const cmsis_nn_transpose_params transpose_params = {TRANSPOSE_NHCW_PERM_SIZE, perm}; in transpose_nhcw_arm_transpose_s8() local
74 …us result = arm_transpose_s8(input_data, output_ptr, &input_dims, &output_dims, &transpose_params); in transpose_nhcw_arm_transpose_s8()
94 const cmsis_nn_transpose_params transpose_params = {TRANSPOSE_WCHN_PERM_SIZE, perm}; in transpose_wchn_arm_transpose_s8() local
96 …us result = arm_transpose_s8(input_data, output_ptr, &input_dims, &output_dims, &transpose_params); in transpose_wchn_arm_transpose_s8()
116 const cmsis_nn_transpose_params transpose_params = {TRANSPOSE_NCHW_PERM_SIZE, perm}; in transpose_nchw_arm_transpose_s8() local
118 …us result = arm_transpose_s8(input_data, output_ptr, &input_dims, &output_dims, &transpose_params); in transpose_nchw_arm_transpose_s8()
138 const cmsis_nn_transpose_params transpose_params = {TRANSPOSE_CHWN_PERM_SIZE, perm}; in transpose_chwn_arm_transpose_s8() local
140 …us result = arm_transpose_s8(input_data, output_ptr, &input_dims, &output_dims, &transpose_params); in transpose_chwn_arm_transpose_s8()
[all …]
|
| /cmsis-nn-latest/Source/TransposeFunctions/ |
| D | arm_transpose_s8.c | 170 const cmsis_nn_transpose_params *const transpose_params) in arm_transpose_s8() argument
175 const uint32_t *const perm = transpose_params->permutations; in arm_transpose_s8()
187 if (transpose_params->num_dims == 1) in arm_transpose_s8()
193 else if (transpose_params->num_dims == 2) in arm_transpose_s8()
199 else if (transpose_params->num_dims == 3) in arm_transpose_s8()
|
| /cmsis-nn-latest/Source/ConvolutionFunctions/ |
| D | arm_depthwise_conv_wrapper_s8.c | 67 const cmsis_nn_transpose_params transpose_params = {4, perm}; in arm_depthwise_conv_to_conv_s8() local
69 …atus status = arm_transpose_s8(filter, w_buf, filter_dims, &filter_output_dims, &transpose_params); in arm_depthwise_conv_to_conv_s8()
|
| /cmsis-nn-latest/Include/ |
| D | arm_nnfunctions.h | 2431 const cmsis_nn_transpose_params *const transpose_params);
|
