Home
last modified time | relevance | path

Searched refs:input_dims (Results 1 – 25 of 43) sorted by relevance

12

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/third_party/cmsis/CMSIS/NN/Source/ConvolutionFunctions/
A Darm_convolve_wrapper_s8.c53 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 Darm_convolve_1x1_s8_fast.c56 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 Darm_depthwise_conv_wrapper_s8.c52 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 Darm_convolve_1_x_n_s8.c53 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 Darm_convolve_s8.c54 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 Darm_depthwise_conv_s8_opt.c54 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 Darm_convolve_wrapper_s8.c53 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 Darm_convolve_1x1_s8_fast.c56 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 Darm_depthwise_conv_wrapper_s8.c52 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 Darm_convolve_1_x_n_s8.c53 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 Darm_convolve_s8.c54 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 Darm_depthwise_conv_s8_opt.c54 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 Dlegacy_reference_ops.h521 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 Dreduce.h34 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 Dexpand_dims.cc32 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 Dsplit.cc31 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 Dsplit_v.cc31 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 Dsqueeze.cc48 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 Dunpack.cc36 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 Dconv.cc66 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 Ddepthwise_conv.cc118 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 Dpooling.cc131 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 Darm_nnfunctions.h197 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 Darm_nnfunctions.h197 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 Dlegacy_optimized_ops.h2338 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 …]

Completed in 3274 milliseconds

12