Lines Matching refs:input_dims
57 cmsis_nn_dims input_dims; in ZTEST() local
61 input_dims.n = AVGPOOLING_2_INPUT_BATCHES; in ZTEST()
62 input_dims.w = AVGPOOLING_2_INPUT_W; in ZTEST()
63 input_dims.h = AVGPOOLING_2_INPUT_H; in ZTEST()
64 input_dims.c = AVGPOOLING_2_IN_CH; in ZTEST()
84 &input_dims, in ZTEST()
158 cmsis_nn_dims input_dims; in ZTEST() local
167 input_dims.n = CONV_4_INPUT_BATCHES; in ZTEST()
168 input_dims.w = CONV_4_INPUT_W; in ZTEST()
169 input_dims.h = CONV_4_INPUT_H; in ZTEST()
170 input_dims.c = CONV_4_IN_CH; in ZTEST()
192 int32_t buf_size = arm_convolve_s8_get_buffer_size(&input_dims, &filter_dims); in ZTEST()
200 &input_dims, in ZTEST()
213 buf_size = arm_convolve_wrapper_s8_get_buffer_size(&conv_params, &input_dims, in ZTEST()
221 &input_dims, in ZTEST()
280 cmsis_nn_dims input_dims; in ZTEST() local
289 input_dims.n = STRIDE2PAD1_INPUT_BATCHES; in ZTEST()
290 input_dims.w = STRIDE2PAD1_INPUT_W; in ZTEST()
291 input_dims.h = STRIDE2PAD1_INPUT_H; in ZTEST()
292 input_dims.c = STRIDE2PAD1_IN_CH; in ZTEST()
321 &input_dims, in ZTEST()
376 cmsis_nn_dims input_dims; in ZTEST() local
385 input_dims.n = FULLY_CONNECTED_MVE_0_INPUT_BATCHES; in ZTEST()
386 input_dims.w = FULLY_CONNECTED_MVE_0_INPUT_W; in ZTEST()
387 input_dims.h = FULLY_CONNECTED_MVE_0_INPUT_H; in ZTEST()
388 input_dims.c = FULLY_CONNECTED_MVE_0_IN_CH; in ZTEST()
410 &input_dims, in ZTEST()
460 cmsis_nn_dims input_dims; in ZTEST() local
466 input_dims.n = MAXPOOLING_2_INPUT_BATCHES; in ZTEST()
467 input_dims.w = MAXPOOLING_2_INPUT_W; in ZTEST()
468 input_dims.h = MAXPOOLING_2_INPUT_H; in ZTEST()
469 input_dims.c = MAXPOOLING_2_IN_CH; in ZTEST()
486 arm_max_pool_s8(&ctx, &pool_params, &input_dims, input_data, &filter_dims, in ZTEST()
580 cmsis_nn_dims input_dims; in ZTEST() local
593 input_dims.n = SVDF_2_INPUT_BATCHES; in ZTEST()
594 input_dims.h = SVDF_2_INPUT_SIZE; in ZTEST()
635 &input_dims, in ZTEST()