Home
last modified time | relevance | path

Searched refs:op_params (Results 1 – 25 of 39) sorted by relevance

12

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/
A Dlegacy_reference_ops.h724 op_params.axis = 3 - axis; in TensorFlowSplit()
873 op_params.scale = scale; in Dequantize()
898 op_params.batch_dims = 0; in Gather()
960 op_params.axis[i] = reduction_indices[op_params.axis_count - 1 - i]; in Mean()
1150 op_params.axis = 3 - dim; in Pack()
1182 op_params.axis = 3 - dim; in Pack()
1957 op_params.range = range; in LocalResponseNormalization()
1958 op_params.bias = bias; in LocalResponseNormalization()
1959 op_params.alpha = alpha; in LocalResponseNormalization()
1960 op_params.beta = beta; in LocalResponseNormalization()
[all …]
A Dpad.h38 inline void PadImpl(const tflite::PadParams& op_params, in PadImpl() argument
55 for (int i = 0; i < op_params.left_padding_count; ++i) { in PadImpl()
57 op_params.left_padding_count] = op_params.left_padding[i]; in PadImpl()
63 for (int i = 0; i < op_params.right_padding_count; ++i) { in PadImpl()
65 op_params.right_padding_count] = in PadImpl()
66 op_params.right_padding[i]; in PadImpl()
118 inline void Pad(const tflite::PadParams& op_params, in Pad() argument
128 inline void Pad(const tflite::PadParams& op_params, in Pad() argument
139 inline void Pad(const tflite::PadParams& op_params, in Pad() argument
148 inline void PadImageStyle(const tflite::PadParams& op_params, in PadImageStyle() argument
[all …]
A Dcomparisons.h84 int left_shift = op_params.left_shift; in ComparisonWithScaling()
85 int32_t input1_offset = op_params.input1_offset; in ComparisonWithScaling()
87 int input1_shift = op_params.input1_shift; in ComparisonWithScaling()
88 int32_t input2_offset = op_params.input2_offset; in ComparisonWithScaling()
90 int input2_shift = op_params.input2_shift; in ComparisonWithScaling()
132 const ComparisonParams& op_params, in BroadcastComparison4DSlowImpl() argument
169 const ComparisonParams& op_params, in BroadcastComparison4DSlowWithScaling() argument
178 int left_shift = op_params.left_shift; in BroadcastComparison4DSlowWithScaling()
179 int32_t input1_offset = op_params.input1_offset; in BroadcastComparison4DSlowWithScaling()
181 int input1_shift = op_params.input1_shift; in BroadcastComparison4DSlowWithScaling()
[all …]
A Ddequantize.h31 inline void Dequantize(const tflite::DequantizationParams& op_params, in Dequantize() argument
35 int32_t zero_point = op_params.zero_point; in Dequantize()
36 const double scale = op_params.scale; in Dequantize()
49 const tflite::PerChannelDequantizationParams& op_params, in PerChannelDequantize() argument
55 const int32_t* zero_point = op_params.zero_point; in PerChannelDequantize()
56 const float* scale = op_params.scale; in PerChannelDequantize()
57 const int32_t quantized_dimension = op_params.quantized_dimension; in PerChannelDequantize()
A Dresize_bilinear.h47 inline void ResizeBilinear(const tflite::ResizeBilinearParams& op_params, in ResizeBilinear() argument
55 TFLITE_DCHECK(!op_params.half_pixel_centers || !op_params.align_corners); in ResizeBilinear()
82 if (op_params.align_corners && output_height > 1) { in ResizeBilinear()
85 if (op_params.align_corners && output_width > 1) { in ResizeBilinear()
94 ComputeInterpolationValues(y, height_scale, op_params.half_pixel_centers, in ResizeBilinear()
137 const tflite::ResizeBilinearParams& op_params, in ResizeBilinearInteger() argument
143 TFLITE_DCHECK(!op_params.half_pixel_centers || !op_params.align_corners); in ResizeBilinearInteger()
172 if (op_params.align_corners && output_height > 1) { in ResizeBilinearInteger()
177 if (op_params.align_corners && output_width > 1) { in ResizeBilinearInteger()
186 op_params.half_pixel_centers, in ResizeBilinearInteger()
[all …]
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/
A Ddepthwise_conv_common.cc44 DepthwiseParams op_params; in DepthwiseConvParamsFloat() local
50 op_params.stride_width = params.stride_width; in DepthwiseConvParamsFloat()
51 op_params.stride_height = params.stride_height; in DepthwiseConvParamsFloat()
55 return op_params; in DepthwiseConvParamsFloat()
62 DepthwiseParams op_params; in DepthwiseConvParamsQuantized() local
63 op_params.input_offset = -data.input_zero_point; in DepthwiseConvParamsQuantized()
65 op_params.output_offset = data.output_zero_point; in DepthwiseConvParamsQuantized()
67 op_params.output_shift = -data.output_shift; in DepthwiseConvParamsQuantized()
71 op_params.stride_height = params.stride_height; in DepthwiseConvParamsQuantized()
72 op_params.stride_width = params.stride_width; in DepthwiseConvParamsQuantized()
[all …]
A Dconv_common.cc43 ConvParams op_params; in ConvParamsFloat() local
49 op_params.stride_width = params.stride_width; in ConvParamsFloat()
50 op_params.stride_height = params.stride_height; in ConvParamsFloat()
53 return op_params; in ConvParamsFloat()
60 ConvParams op_params; in ConvParamsQuantized() local
61 op_params.input_offset = -data.input_zero_point; in ConvParamsQuantized()
63 op_params.output_offset = data.output_zero_point; in ConvParamsQuantized()
65 op_params.output_shift = -data.output_shift; in ConvParamsQuantized()
69 op_params.stride_height = params.stride_height; in ConvParamsQuantized()
70 op_params.stride_width = params.stride_width; in ConvParamsQuantized()
[all …]
A Dfully_connected_common.cc36 FullyConnectedParams op_params; in FullyConnectedParamsQuantized() local
37 op_params.input_offset = -op_data.input_zero_point; in FullyConnectedParamsQuantized()
38 op_params.weights_offset = -op_data.filter_zero_point; in FullyConnectedParamsQuantized()
39 op_params.output_offset = op_data.output_zero_point; in FullyConnectedParamsQuantized()
40 op_params.output_multiplier = op_data.output_multiplier; in FullyConnectedParamsQuantized()
41 op_params.output_shift = op_data.output_shift; in FullyConnectedParamsQuantized()
42 op_params.quantized_activation_min = op_data.output_activation_min; in FullyConnectedParamsQuantized()
44 return op_params; in FullyConnectedParamsQuantized()
49 FullyConnectedParams op_params; in FullyConnectedParamsFloat() local
51 &op_params.float_activation_max); in FullyConnectedParamsFloat()
[all …]
A Dstrided_slice.cc63 tflite::StridedSliceParams op_params; in BuildStridedSliceParams() local
64 op_params.start_indices_count = op_context->dims; in BuildStridedSliceParams()
65 op_params.stop_indices_count = op_context->dims; in BuildStridedSliceParams()
66 op_params.strides_count = op_context->dims; in BuildStridedSliceParams()
75 op_params.ellipsis_mask = 0; in BuildStridedSliceParams()
77 op_params.new_axis_mask = 0; in BuildStridedSliceParams()
79 return op_params; in BuildStridedSliceParams()
127 StridedSliceParams* op_params = in Prepare() local
141 const StridedSliceParams& op_params = in Eval() local
150 reference_ops::StridedSlice(op_params, in Eval()
[all …]
A Dsub.cc128 tflite::ArithmeticParams op_params; in EvalSub() local
132 op_params, tflite::micro::GetTensorShape(input1), in EvalSub()
140 op_params, tflite::micro::GetTensorShape(input1), in EvalSub()
155 tflite::ArithmeticParams op_params; in EvalSubQuantized() local
156 op_params.left_shift = data->left_shift; in EvalSubQuantized()
157 op_params.input1_offset = data->input1_offset; in EvalSubQuantized()
159 op_params.input1_shift = data->input1_shift; in EvalSubQuantized()
160 op_params.input2_offset = data->input2_offset; in EvalSubQuantized()
162 op_params.input2_shift = data->input2_shift; in EvalSubQuantized()
163 op_params.output_offset = data->output_offset; in EvalSubQuantized()
[all …]
A Dl2_pool_2d.cc93 op_params->float_activation_min = activation_min; in L2EvalFloat()
94 op_params->float_activation_max = activation_max; in L2EvalFloat()
95 reference_ops::L2Pool(*op_params, tflite::micro::GetTensorShape(&input), in L2EvalFloat()
109 tflite::PoolParams op_params; in L2Eval() local
110 op_params.stride_height = params->stride_height; in L2Eval()
111 op_params.stride_width = params->stride_width; in L2Eval()
112 op_params.filter_height = params->filter_height; in L2Eval()
113 op_params.filter_width = params->filter_width; in L2Eval()
114 op_params.padding_values.height = params->computed.padding.height; in L2Eval()
115 op_params.padding_values.width = params->computed.padding.width; in L2Eval()
[all …]
A Dleaky_relu.cc46 LeakyReluParams op_params = {}; in QuantizeLeakyRelu() local
48 op_params.input_offset = data.input_zero_point; in QuantizeLeakyRelu()
49 op_params.output_offset = data.output_zero_point; in QuantizeLeakyRelu()
50 op_params.output_multiplier_alpha = data.output_multiplier_alpha; in QuantizeLeakyRelu()
51 op_params.output_shift_alpha = data.output_shift_alpha; in QuantizeLeakyRelu()
52 op_params.output_multiplier_identity = data.output_multiplier_identity; in QuantizeLeakyRelu()
53 op_params.output_shift_identity = data.output_shift_identity; in QuantizeLeakyRelu()
54 reference_ops::QuantizeLeakyRelu(op_params, in QuantizeLeakyRelu()
115 LeakyReluParams op_params = {}; in LeakyReluEval() local
119 op_params.alpha = params->alpha; in LeakyReluEval()
[all …]
A Dresize_bilinear.cc73 tflite::ResizeBilinearParams op_params; in Eval() local
74 op_params.align_corners = params->align_corners; in Eval()
75 op_params.half_pixel_centers = params->half_pixel_centers; in Eval()
76 reference_ops::ResizeBilinear(op_params, in Eval()
84 tflite::ResizeBilinearParams op_params; in Eval() local
85 op_params.align_corners = params->align_corners; in Eval()
86 op_params.half_pixel_centers = params->half_pixel_centers; in Eval()
88 op_params, tflite::micro::GetTensorShape(input), in Eval()
A Dlog_softmax.cc108 SoftmaxParams op_params = {}; in LogSoftmaxEval() local
109 reference_ops::LogSoftmax(op_params, tflite::micro::GetTensorShape(input), in LogSoftmaxEval()
116 SoftmaxParams op_params = {}; in LogSoftmaxEval() local
117 op_params.input_multiplier = data->input_multiplier; in LogSoftmaxEval()
118 op_params.input_left_shift = data->input_left_shift; in LogSoftmaxEval()
119 op_params.reverse_scaling_divisor = data->reverse_scaling_divisor; in LogSoftmaxEval()
120 op_params.reverse_scaling_right_shift = data->reverse_scaling_right_shift; in LogSoftmaxEval()
121 op_params.diff_min = data->diff_min; in LogSoftmaxEval()
122 reference_ops::LogSoftmax(op_params, data->outer_size, data->depth, in LogSoftmaxEval()
A Dresize_nearest_neighbor.cc69 tflite::ResizeNearestNeighborParams op_params; in Eval() local
70 op_params.align_corners = params->align_corners; in Eval()
71 op_params.half_pixel_centers = false; in Eval()
75 op_params, tflite::micro::GetTensorShape(input), in Eval()
83 op_params, tflite::micro::GetTensorShape(input), in Eval()
91 op_params, tflite::micro::GetTensorShape(input), in Eval()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/cmsis_nn/
A Dadd.cc110 tflite::ArithmeticParams op_params; in EvalAdd() local
134 tflite::ArithmeticParams op_params; in EvalAddQuantized() local
135 op_params.left_shift = data->left_shift; in EvalAddQuantized()
138 op_params.input1_shift = data->input1_shift; in EvalAddQuantized()
141 op_params.input2_shift = data->input2_shift; in EvalAddQuantized()
164 op_params.input1_offset, op_params.input1_multiplier, in EvalAddQuantized()
165 op_params.input1_shift, op_params.input2_offset, in EvalAddQuantized()
166 op_params.input2_multiplier, op_params.input2_shift, in EvalAddQuantized()
168 op_params.output_offset, op_params.output_multiplier, in EvalAddQuantized()
169 op_params.output_shift, op_params.quantized_activation_min, in EvalAddQuantized()
[all …]
A Dpooling.cc86 PoolParams op_params; in AverageEvalFloat() local
87 op_params.stride_height = params->stride_height; in AverageEvalFloat()
88 op_params.stride_width = params->stride_width; in AverageEvalFloat()
90 op_params.filter_width = params->filter_width; in AverageEvalFloat()
108 PoolParams op_params; in AverageEvalQuantized() local
179 tflite::PoolParams op_params; in MaxEvalFloat() local
181 op_params.stride_width = params->stride_width; in MaxEvalFloat()
183 op_params.filter_width = params->filter_width; in MaxEvalFloat()
198 tflite::PoolParams op_params; in MaxEvalQuantizedUInt8() local
200 op_params.stride_width = params->stride_width; in MaxEvalQuantizedUInt8()
[all …]
A Dmul.cc107 tflite::ArithmeticParams op_params; in EvalQuantized() local
110 op_params.input1_offset = -data.input1_zero_point; in EvalQuantized()
111 op_params.input2_offset = -data.input2_zero_point; in EvalQuantized()
112 op_params.output_offset = data.output_zero_point; in EvalQuantized()
113 op_params.output_multiplier = data.output_multiplier; in EvalQuantized()
114 op_params.output_shift = data.output_shift; in EvalQuantized()
118 tflite::micro::GetTensorShape(input2), &op_params); in EvalQuantized()
136 op_params.output_offset, op_params.output_multiplier, in EvalQuantized()
137 op_params.output_shift, op_params.quantized_activation_min, in EvalQuantized()
138 op_params.quantized_activation_max, in EvalQuantized()
[all …]
A Ddepthwise_conv.cc262 tflite::DepthwiseParams op_params; in EvalQuantized() local
264 op_params.padding_type = PaddingType::kSame; in EvalQuantized()
265 op_params.padding_values.width = data.padding.width; in EvalQuantized()
267 op_params.stride_width = params->stride_width; in EvalQuantized()
268 op_params.stride_height = params->stride_height; in EvalQuantized()
274 op_params.input_offset = input_offset; in EvalQuantized()
275 op_params.weights_offset = filter_offset; in EvalQuantized()
276 op_params.output_offset = output_offset; in EvalQuantized()
277 op_params.output_multiplier = data.output_multiplier; in EvalQuantized()
279 op_params.output_shift = -data.output_shift; in EvalQuantized()
[all …]
A Dconv.cc256 ConvParams op_params; in EvalQuantized() local
258 op_params.padding_values.width = data.padding.width; in EvalQuantized()
259 op_params.padding_values.height = data.padding.height; in EvalQuantized()
260 op_params.stride_width = params->stride_width; in EvalQuantized()
261 op_params.stride_height = params->stride_height; in EvalQuantized()
262 op_params.dilation_width_factor = params->dilation_width_factor; in EvalQuantized()
264 op_params.input_offset = input_offset; in EvalQuantized()
265 op_params.weights_offset = filter_offset; in EvalQuantized()
266 op_params.output_offset = output_offset; in EvalQuantized()
267 op_params.output_multiplier = data.output_multiplier; in EvalQuantized()
[all …]
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/
A Dstrided_slice_logic.h186 tflite::StridedSliceParams op_params; in BuildStridedSliceParams() local
189 op_params.start_indices_count = dims_count; in BuildStridedSliceParams()
190 op_params.stop_indices_count = dims_count; in BuildStridedSliceParams()
191 op_params.strides_count = dims_count; in BuildStridedSliceParams()
194 op_params.stop_indices[i] = stop_indices[i]; in BuildStridedSliceParams()
195 op_params.strides[i] = strides[i]; in BuildStridedSliceParams()
198 op_params.begin_mask = begin_mask; in BuildStridedSliceParams()
199 op_params.ellipsis_mask = 0; in BuildStridedSliceParams()
200 op_params.end_mask = end_mask; in BuildStridedSliceParams()
201 op_params.new_axis_mask = 0; in BuildStridedSliceParams()
[all …]
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/
A Ddequantize.h44 DequantizationParams op_params; in DequantizeImpl() local
45 op_params.zero_point = input->params.zero_point; in DequantizeImpl()
46 op_params.scale = input->params.scale; in DequantizeImpl()
51 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in DequantizeImpl()
55 op_params, GetTensorShape(input), GetTensorData<uint8_t>(input), in DequantizeImpl()
62 op_params, GetTensorShape(input), GetTensorData<int8_t>(input), in DequantizeImpl()
66 op_params, GetTensorShape(input), GetTensorData<int8_t>(input), in DequantizeImpl()
73 op_params, GetTensorShape(input), GetTensorData<int16_t>(input), in DequantizeImpl()
77 op_params, GetTensorShape(input), GetTensorData<int16_t>(input), in DequantizeImpl()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/integer_ops/
A Dmean.h26 inline void MeanImpl(const tflite::MeanParams& op_params, in MeanImpl() argument
41 TFLITE_CHECK_EQ(op_params.axis_count, 2); in MeanImpl()
42 TFLITE_CHECK((op_params.axis[0] == 1 && op_params.axis[1] == 2) || in MeanImpl()
43 (op_params.axis[0] == 2 && op_params.axis[1] == 1)); in MeanImpl()
144 MeanWorkerTask(const tflite::MeanParams& op_params, in MeanWorkerTask()
149 : op_params(op_params), in MeanWorkerTask()
166 const tflite::MeanParams& op_params;
178 inline void Mean(const tflite::MeanParams& op_params, in Mean() argument
197 TFLITE_CHECK_EQ(op_params.axis_count, 2); in Mean()
198 TFLITE_CHECK((op_params.axis[0] == 1 && op_params.axis[1] == 2) || in Mean()
[all …]
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/
A Dlegacy_optimized_ops.h2378 tflite::ConvParams op_params; in DilatedIm2col() local
2398 tflite::ConvParams op_params; in Im2col() local
2532 tflite::ConvParams op_params; in Conv() local
4776 op_params.range = range; in LocalResponseNormalization()
4777 op_params.bias = bias; in LocalResponseNormalization()
4778 op_params.alpha = alpha; in LocalResponseNormalization()
4779 op_params.beta = beta; in LocalResponseNormalization()
4860 tflite::PadParams op_params; in PadV2() local
4900 op_params.begin_count = 4; in Slice()
4901 op_params.size_count = 4; in Slice()
[all …]
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/integer_ops/
A Dmean.h24 inline void Mean(const tflite::MeanParams& op_params, int32_t multiplier, in Mean() argument
45 TFLITE_CHECK_EQ(op_params.axis_count, 2); in Mean()
46 TFLITE_CHECK((op_params.axis[0] == 1 && op_params.axis[1] == 2) || in Mean()
47 (op_params.axis[0] == 2 && op_params.axis[1] == 1)); in Mean()

Completed in 60 milliseconds

12