32 cmsis_nn_dims input_dims;
33 cmsis_nn_dims output_dims;
34 cmsis_nn_pool_params pool_params;
35 cmsis_nn_dims filter_dims;
38 const int depth = input_shape.
dims(3);
42 input_dims.h = input_shape.
dims(1);
43 input_dims.w = input_shape.
dims(2);
48 output_dims.w = output_width;
49 output_dims.c = depth;
63 if (data_type == luci_interpreter::DataType::S8)
65 arm_max_pool_s8(&ctx, &pool_params, &input_dims,
66 luci_interpreter::kernels::getTensorData<int8_t>(input_data), &filter_dims,
67 &output_dims, luci_interpreter::kernels::getTensorData<int8_t>(output_data));
71 arm_max_pool_s16(&ctx, &pool_params, &input_dims,
72 luci_interpreter::kernels::getTensorData<int16_t>(input_data), &filter_dims,
73 &output_dims, luci_interpreter::kernels::getTensorData<int16_t>(output_data));
void MaxPool(const PoolParams ¶ms, const luci_interpreter::RuntimeShape &input_shape, const uint8_t *input_data, const luci_interpreter::RuntimeShape &output_shape, uint8_t *output_data, luci_interpreter::DataType data_type)