/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/third_party/cmsis/CMSIS/NN/Source/ConvolutionFunctions/ |
A D | arm_convolve_wrapper_s8.c | 53 const cmsis_nn_dims *input_dims, in arm_convolve_wrapper_s8() argument 62 … if ((conv_params->padding.w == 0) && (conv_params->padding.h == 0) && (input_dims->c % 4 == 0) && in arm_convolve_wrapper_s8() 68 input_dims, in arm_convolve_wrapper_s8() 78 (input_dims->n == 1)) in arm_convolve_wrapper_s8() 83 input_dims, in arm_convolve_wrapper_s8() 97 input_dims, in arm_convolve_wrapper_s8() 109 const cmsis_nn_dims *input_dims, in arm_convolve_wrapper_s8_get_buffer_size() argument 116 return arm_convolve_1x1_s8_fast_get_buffer_size(input_dims); in arm_convolve_wrapper_s8_get_buffer_size() 119 (input_dims->n == 1)) in arm_convolve_wrapper_s8_get_buffer_size() 121 return arm_convolve_1_x_n_s8_get_buffer_size(input_dims, filter_dims); in arm_convolve_wrapper_s8_get_buffer_size() [all …]
|
A D | arm_convolve_1x1_s8_fast.c | 56 const cmsis_nn_dims *input_dims, in arm_convolve_1x1_s8_fast() argument 65 if (input_dims->c % 4 != 0 || conv_params->padding.w != 0 || conv_params->padding.h != 0 || in arm_convolve_1x1_s8_fast() 77 const int32_t col_len = input_dims->w * input_dims->h * input_dims->n; in arm_convolve_1x1_s8_fast() 79 const int32_t input_ch = input_dims->c; in arm_convolve_1x1_s8_fast() 149 const int32_t lhs_rows = input_dims->w * input_dims->h * input_dims->n; in arm_convolve_1x1_s8_fast() 151 const int32_t rhs_cols = input_dims->c; in arm_convolve_1x1_s8_fast() 173 int32_t arm_convolve_1x1_s8_fast_get_buffer_size(const cmsis_nn_dims *input_dims) in arm_convolve_1x1_s8_fast_get_buffer_size() argument 175 (void)input_dims; in arm_convolve_1x1_s8_fast_get_buffer_size()
|
A D | arm_depthwise_conv_wrapper_s8.c | 52 const cmsis_nn_dims *input_dims, in arm_depthwise_conv_wrapper_s8() argument 70 input_dims, in arm_depthwise_conv_wrapper_s8() 85 input_dims, in arm_depthwise_conv_wrapper_s8() 100 input_dims, in arm_depthwise_conv_wrapper_s8() 115 const cmsis_nn_dims *input_dims, in arm_depthwise_conv_wrapper_s8_get_buffer_size() argument 122 if (input_dims->c == output_dims->c) in arm_depthwise_conv_wrapper_s8_get_buffer_size() 124 size = arm_depthwise_conv_s8_opt_get_buffer_size(input_dims, filter_dims); in arm_depthwise_conv_wrapper_s8_get_buffer_size()
|
A D | arm_convolve_1_x_n_s8.c | 53 const cmsis_nn_dims *input_dims, in arm_convolve_1_x_n_s8() argument 73 const uint16_t input_x = input_dims->w; in arm_convolve_1_x_n_s8() 77 const uint16_t input_ch = input_dims->c; in arm_convolve_1_x_n_s8() 175 input_dims, in arm_convolve_1_x_n_s8() 190 int32_t arm_convolve_1_x_n_s8_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_dims … in arm_convolve_1_x_n_s8_get_buffer_size() argument 193 return (2 * input_dims->c * filter_dims->w * filter_dims->h) * sizeof(int16_t); in arm_convolve_1_x_n_s8_get_buffer_size() 195 (void)input_dims; in arm_convolve_1_x_n_s8_get_buffer_size()
|
A D | arm_convolve_s8.c | 54 const cmsis_nn_dims *input_dims, in arm_convolve_s8() argument 66 const uint16_t input_batches = input_dims->n; in arm_convolve_s8() 67 const uint16_t input_x = input_dims->w; in arm_convolve_s8() 68 const uint16_t input_y = input_dims->h; in arm_convolve_s8() 69 const uint16_t input_ch = input_dims->c; in arm_convolve_s8() 367 int32_t arm_convolve_s8_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filte… in arm_convolve_s8_get_buffer_size() argument 370 return (2 * input_dims->c * filter_dims->w * filter_dims->h) * (int32_t)sizeof(int16_t); in arm_convolve_s8_get_buffer_size() 372 (void)input_dims; in arm_convolve_s8_get_buffer_size()
|
A D | arm_depthwise_conv_s8_opt.c | 54 const cmsis_nn_dims *input_dims, in arm_depthwise_conv_s8_opt() argument 64 const int32_t input_ch = input_dims->c; in arm_depthwise_conv_s8_opt() 73 const int32_t input_x = input_dims->w; in arm_depthwise_conv_s8_opt() 74 const int32_t input_y = input_dims->h; in arm_depthwise_conv_s8_opt() 398 input_dims, in arm_depthwise_conv_s8_opt() 412 int32_t arm_depthwise_conv_s8_opt_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_d… in arm_depthwise_conv_s8_opt_get_buffer_size() argument 416 return (2 * input_dims->c * filter_dims->w * filter_dims->h) * (int32_t)sizeof(int16_t) + 4; in arm_depthwise_conv_s8_opt_get_buffer_size() 418 return (input_dims->c * filter_dims->w * filter_dims->h) * sizeof(int16_t); in arm_depthwise_conv_s8_opt_get_buffer_size() 420 (void)input_dims; in arm_depthwise_conv_s8_opt_get_buffer_size()
|
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/tools/make/downloads/cmsis/CMSIS/NN/Source/ConvolutionFunctions/ |
A D | arm_convolve_wrapper_s8.c | 53 const cmsis_nn_dims *input_dims, in arm_convolve_wrapper_s8() argument 62 … if ((conv_params->padding.w == 0) && (conv_params->padding.h == 0) && (input_dims->c % 4 == 0) && in arm_convolve_wrapper_s8() 68 input_dims, in arm_convolve_wrapper_s8() 78 (input_dims->n == 1)) in arm_convolve_wrapper_s8() 83 input_dims, in arm_convolve_wrapper_s8() 97 input_dims, in arm_convolve_wrapper_s8() 109 const cmsis_nn_dims *input_dims, in arm_convolve_wrapper_s8_get_buffer_size() argument 116 return arm_convolve_1x1_s8_fast_get_buffer_size(input_dims); in arm_convolve_wrapper_s8_get_buffer_size() 119 (input_dims->n == 1)) in arm_convolve_wrapper_s8_get_buffer_size() 121 return arm_convolve_1_x_n_s8_get_buffer_size(input_dims, filter_dims); in arm_convolve_wrapper_s8_get_buffer_size() [all …]
|
A D | arm_convolve_1x1_s8_fast.c | 56 const cmsis_nn_dims *input_dims, in arm_convolve_1x1_s8_fast() argument 65 if (input_dims->c % 4 != 0 || conv_params->padding.w != 0 || conv_params->padding.h != 0 || in arm_convolve_1x1_s8_fast() 77 const int32_t col_len = input_dims->w * input_dims->h * input_dims->n; in arm_convolve_1x1_s8_fast() 79 const int32_t input_ch = input_dims->c; in arm_convolve_1x1_s8_fast() 149 const int32_t lhs_rows = input_dims->w * input_dims->h * input_dims->n; in arm_convolve_1x1_s8_fast() 151 const int32_t rhs_cols = input_dims->c; in arm_convolve_1x1_s8_fast() 173 int32_t arm_convolve_1x1_s8_fast_get_buffer_size(const cmsis_nn_dims *input_dims) in arm_convolve_1x1_s8_fast_get_buffer_size() argument 175 (void)input_dims; in arm_convolve_1x1_s8_fast_get_buffer_size()
|
A D | arm_depthwise_conv_wrapper_s8.c | 52 const cmsis_nn_dims *input_dims, in arm_depthwise_conv_wrapper_s8() argument 70 input_dims, in arm_depthwise_conv_wrapper_s8() 85 input_dims, in arm_depthwise_conv_wrapper_s8() 100 input_dims, in arm_depthwise_conv_wrapper_s8() 115 const cmsis_nn_dims *input_dims, in arm_depthwise_conv_wrapper_s8_get_buffer_size() argument 122 if (input_dims->c == output_dims->c) in arm_depthwise_conv_wrapper_s8_get_buffer_size() 124 size = arm_depthwise_conv_s8_opt_get_buffer_size(input_dims, filter_dims); in arm_depthwise_conv_wrapper_s8_get_buffer_size()
|
A D | arm_convolve_1_x_n_s8.c | 53 const cmsis_nn_dims *input_dims, in arm_convolve_1_x_n_s8() argument 73 const uint16_t input_x = input_dims->w; in arm_convolve_1_x_n_s8() 77 const uint16_t input_ch = input_dims->c; in arm_convolve_1_x_n_s8() 175 input_dims, in arm_convolve_1_x_n_s8() 190 int32_t arm_convolve_1_x_n_s8_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_dims … in arm_convolve_1_x_n_s8_get_buffer_size() argument 193 return (2 * input_dims->c * filter_dims->w * filter_dims->h) * sizeof(int16_t); in arm_convolve_1_x_n_s8_get_buffer_size() 195 (void)input_dims; in arm_convolve_1_x_n_s8_get_buffer_size()
|
A D | arm_convolve_s8.c | 54 const cmsis_nn_dims *input_dims, in arm_convolve_s8() argument 66 const uint16_t input_batches = input_dims->n; in arm_convolve_s8() 67 const uint16_t input_x = input_dims->w; in arm_convolve_s8() 68 const uint16_t input_y = input_dims->h; in arm_convolve_s8() 69 const uint16_t input_ch = input_dims->c; in arm_convolve_s8() 367 int32_t arm_convolve_s8_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_dims *filte… in arm_convolve_s8_get_buffer_size() argument 370 return (2 * input_dims->c * filter_dims->w * filter_dims->h) * (int32_t)sizeof(int16_t); in arm_convolve_s8_get_buffer_size() 372 (void)input_dims; in arm_convolve_s8_get_buffer_size()
|
A D | arm_depthwise_conv_s8_opt.c | 54 const cmsis_nn_dims *input_dims, in arm_depthwise_conv_s8_opt() argument 64 const int32_t input_ch = input_dims->c; in arm_depthwise_conv_s8_opt() 73 const int32_t input_x = input_dims->w; in arm_depthwise_conv_s8_opt() 74 const int32_t input_y = input_dims->h; in arm_depthwise_conv_s8_opt() 398 input_dims, in arm_depthwise_conv_s8_opt() 412 int32_t arm_depthwise_conv_s8_opt_get_buffer_size(const cmsis_nn_dims *input_dims, const cmsis_nn_d… in arm_depthwise_conv_s8_opt_get_buffer_size() argument 416 return (2 * input_dims->c * filter_dims->w * filter_dims->h) * (int32_t)sizeof(int16_t) + 4; in arm_depthwise_conv_s8_opt_get_buffer_size() 418 return (input_dims->c * filter_dims->w * filter_dims->h) * sizeof(int16_t); in arm_depthwise_conv_s8_opt_get_buffer_size() 420 (void)input_dims; in arm_depthwise_conv_s8_opt_get_buffer_size()
|
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/ |
A D | legacy_reference_ops.h | 521 const uint8* input_data, const Dims<4>& input_dims, in ShuffledFullyConnected() argument 657 ShapeFromDims(*input_dims[i], &input_shapes[i]); in Concatenation() 678 ShapeFromDims(*input_dims[i], &input_shapes[i]); in Concatenation() 703 ShapeFromDims(*input_dims[i], &input_shapes[i]); in DepthConcatenation() 727 Split(op_params, DimsToShape(input_dims), input_data, in TensorFlowSplit() 900 Gather(op_params, DimsToShape(input_dims), input_data, in Gather() 1146 ShapeFromDims(*input_dims[i], &input_shapes[i]); in Pack() 1165 Unpack(op_params, DimsToShape(input_dims), input_data, in Unpack() 1178 ShapeFromDims(*input_dims[i], &input_shapes[i]); in Pack() 2050 BatchToSpaceND(DimsToShape(input_dims), input_data, in BatchToSpaceND() [all …]
|
A D | reduce.h | 34 inline bool Reduce(const In* input_data, const int* input_dims, in Reduce() argument 48 size_t output_offset = ReducedOutputOffset(input_num_dims, input_dims, in Reduce() 52 } while (NextIndex(input_num_dims, input_dims, input_iter)); in Reduce() 93 inline bool ReduceSumImpl(const In* input_data, const int* input_dims, in ReduceSumImpl() argument 128 inline bool ReduceGeneric(const T* input_data, const int* input_dims, in ReduceGeneric() argument 137 if (input_dims[i] == 0) return true; in ReduceGeneric() 153 return Reduce<T, T>(input_data, input_dims, output_dims, input_num_dims, in ReduceGeneric() 162 inline bool Mean(const T* input_data, const int* input_dims, in Mean() argument 199 size_t current = static_cast<size_t>(input_dims[resolved_axis[idx]]); in Mean() 324 float input_scale, const int* input_dims, in QuantizedMeanOrSum() argument [all …]
|
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/ |
A D | expand_dims.cc | 32 const TfLiteIntArray* input_dims = input->dims; in ExpandTensorDim() local 35 axis = input_dims->size + 1 + axis; in ExpandTensorDim() 37 TF_LITE_ENSURE(context, (axis <= input_dims->size)); in ExpandTensorDim() 39 output_dims->size = input_dims->size + 1; in ExpandTensorDim() 42 output_dims->data[i] = input_dims->data[i]; in ExpandTensorDim() 46 output_dims->data[i] = input_dims->data[i - 1]; in ExpandTensorDim() 108 const int input_dims = input->dims->size; in Eval() local 113 if ((axis_value > static_cast<int32_t>(input_dims)) || in Eval() 114 (axis_value < static_cast<int32_t>(-(input_dims + 1)))) { in Eval()
|
A D | split.cc | 31 const TfLiteIntArray* input_dims = input->dims; in SplitImpl() local 36 const int split_dimensions = input_dims->size; in SplitImpl() 44 TFLITE_DCHECK_EQ(split_size, input_dims->data[axis]); in SplitImpl() 47 outer_size *= input_dims->data[i]; in SplitImpl() 52 base_inner_size *= input_dims->data[i]; in SplitImpl()
|
A D | split_v.cc | 31 const TfLiteIntArray* input_dims = input->dims; in SplitImpl() local 35 const int split_dimensions = input_dims->size; in SplitImpl() 46 TFLITE_DCHECK_EQ(split_size, input_dims->data[axis_value]); in SplitImpl() 49 outer_size *= input_dims->data[i]; in SplitImpl() 54 base_inner_size *= input_dims->data[i]; in SplitImpl()
|
A D | squeeze.cc | 48 const TfLiteIntArray* input_dims = op_context.input->dims; in Prepare() local 58 if (input_dims->data[idx] == 1) { in Prepare() 67 input_dims->data[current] == 1); in Prepare() 75 TFLITE_CHECK_GE(output_dims->data[out_idx++], input_dims->data[in_idx]); in Prepare()
|
A D | unpack.cc | 36 const TfLiteIntArray* input_dims = input->dims; in UnpackImpl() local 38 const int dimensions = input_dims->size; in UnpackImpl() 48 outer_size *= input_dims->data[i]; in UnpackImpl() 52 copy_size *= input_dims->data[i]; in UnpackImpl()
|
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/cmsis_nn/ |
A D | conv.cc | 66 cmsis_nn_dims input_dims; in Prepare() local 68 input_dims.h = input->dims->data[1]; in Prepare() 69 input_dims.w = input->dims->data[2]; in Prepare() 70 input_dims.c = input_shape.Dims(3); in Prepare() 77 filter_dims.c = input_dims.c; in Prepare() 81 output_dims.n = input_dims.n; in Prepare() 100 context, node, params, input_dims.w, input_dims.h, filter_dims.w, in Prepare() 179 cmsis_nn_dims input_dims; in EvalQuantizedPerChannel() local 180 input_dims.n = batch_size; in EvalQuantizedPerChannel() 181 input_dims.h = input_shape.Dims(1); in EvalQuantizedPerChannel() [all …]
|
A D | depthwise_conv.cc | 118 cmsis_nn_dims input_dims; in Prepare() local 119 input_dims.n = batch_size; in Prepare() 120 input_dims.h = input_height; in Prepare() 121 input_dims.w = input_width; in Prepare() 122 input_dims.c = input_shape.Dims(3); in Prepare() 196 cmsis_nn_dims input_dims; in EvalQuantizedPerChannel() local 197 input_dims.n = batch_size; in EvalQuantizedPerChannel() 198 input_dims.h = input_shape.Dims(1); in EvalQuantizedPerChannel() 199 input_dims.w = input_shape.Dims(2); in EvalQuantizedPerChannel() 200 input_dims.c = input_shape.Dims(3); in EvalQuantizedPerChannel() [all …]
|
A D | pooling.cc | 131 cmsis_nn_dims input_dims; in AverageEvalQuantized() local 132 input_dims.n = 1; in AverageEvalQuantized() 133 input_dims.h = input_shape.Dims(1); in AverageEvalQuantized() 134 input_dims.w = input_shape.Dims(2); in AverageEvalQuantized() 135 input_dims.c = depth; in AverageEvalQuantized() 221 cmsis_nn_dims input_dims; in MaxEvalInt8() local 222 input_dims.n = 1; in MaxEvalInt8() 223 input_dims.h = input_shape.Dims(1); in MaxEvalInt8() 224 input_dims.w = input_shape.Dims(2); in MaxEvalInt8() 225 input_dims.c = depth; in MaxEvalInt8() [all …]
|
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/third_party/cmsis/CMSIS/NN/Include/ |
A D | arm_nnfunctions.h | 197 const cmsis_nn_dims *input_dims, 255 const cmsis_nn_dims *input_dims, 575 const cmsis_nn_dims *input_dims, 632 const cmsis_nn_dims *input_dims, 990 const cmsis_nn_dims *input_dims, 1020 const cmsis_nn_dims *input_dims, 1055 const cmsis_nn_dims *input_dims, 1154 const cmsis_nn_dims *input_dims, 1645 const cmsis_nn_dims *input_dims, 1686 const cmsis_nn_dims *input_dims, [all …]
|
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/tools/make/downloads/cmsis/CMSIS/NN/Include/ |
A D | arm_nnfunctions.h | 197 const cmsis_nn_dims *input_dims, 255 const cmsis_nn_dims *input_dims, 575 const cmsis_nn_dims *input_dims, 632 const cmsis_nn_dims *input_dims, 990 const cmsis_nn_dims *input_dims, 1020 const cmsis_nn_dims *input_dims, 1055 const cmsis_nn_dims *input_dims, 1154 const cmsis_nn_dims *input_dims, 1645 const cmsis_nn_dims *input_dims, 1686 const cmsis_nn_dims *input_dims, [all …]
|
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/ |
A D | legacy_optimized_ops.h | 2338 const uint8* input_data, const Dims<4>& input_dims, in ShuffledFullyConnected() argument 2850 MapAsMatrixWithFirstDimAsRows(input_data, input_dims); in ConvAsGemm() 2878 const int input_rows = input_dims.sizes[0]; in ConvAsGemm() 2879 const int input_cols = FlatSizeSkipDim(input_dims, 0); in ConvAsGemm() 3108 LstmCell(op_params, DimsToShape(input_dims), input_data, in LstmCell() 3424 op_params, DimsToShape(input_dims), input_data_uint8, in LstmCell() 3779 AveragePool(params, DimsToShape(input_dims), input_data, in AveragePool() 3822 AveragePool(params, DimsToShape(input_dims), input_data, in AveragePool() 4447 Logistic(input_data, DimsToShape(input_dims), output_data, in Logistic() 4846 BatchToSpaceND(DimsToShape(input_dims), input_data, in BatchToSpaceND() [all …]
|