Home
last modified time | relevance | path

Searched refs:TfLiteEvalTensor (Results 1 – 25 of 75) sorted by relevance

123

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/
A Dsvdf.h51 const TfLiteEvalTensor* input_tensor,
52 const TfLiteEvalTensor* weights_feature_tensor,
53 const TfLiteEvalTensor* weights_time_tensor,
54 const TfLiteEvalTensor* bias_tensor,
56 TfLiteEvalTensor* activation_state_tensor,
57 TfLiteEvalTensor* output_tensor,
61 TfLiteContext* context, TfLiteNode* node, const TfLiteEvalTensor* input,
62 const TfLiteEvalTensor* weights_feature,
63 const TfLiteEvalTensor* weights_time, const TfLiteEvalTensor* bias,
65 TfLiteEvalTensor* activation_state, TfLiteEvalTensor* output);
A Dkernel_util.h31 inline TfLiteEvalTensor* GetMutableEvalInput(const TfLiteContext* context, in GetMutableEvalInput()
40 inline const TfLiteEvalTensor* GetEvalInput(const TfLiteContext* context, in GetEvalInput()
46 inline TfLiteEvalTensor* GetEvalOutput(const TfLiteContext* context, in GetEvalOutput()
55 T* GetTensorData(TfLiteEvalTensor* tensor) { in GetTensorData()
61 const T* GetTensorData(const TfLiteEvalTensor* tensor) { in GetTensorData()
67 const RuntimeShape GetTensorShape(const TfLiteEvalTensor* tensor);
70 bool HaveSameShapes(const TfLiteEvalTensor* input1,
71 const TfLiteEvalTensor* input2);
81 TfLiteEvalTensor* eval_tensor);
A Dgather_nd.cc105 TfLiteStatus GatherNd(const TfLiteEvalTensor* params, in GatherNd()
106 const TfLiteEvalTensor* indices, in GatherNd()
107 TfLiteEvalTensor* output) { in GatherNd()
151 const TfLiteEvalTensor* params, in EvalGatherNd()
152 const TfLiteEvalTensor* indices, in EvalGatherNd()
153 TfLiteEvalTensor* output) { in EvalGatherNd()
170 const TfLiteEvalTensor* params = in Eval()
172 const TfLiteEvalTensor* indices = in Eval()
174 TfLiteEvalTensor* output = in Eval()
A Dfloor_mod.cc68 const TfLiteEvalTensor* input1, in EvalFloorMod()
69 const TfLiteEvalTensor* input2, in EvalFloorMod()
70 TfLiteEvalTensor* output) { in EvalFloorMod()
93 const TfLiteEvalTensor* input1 = in Eval()
95 const TfLiteEvalTensor* input2 = in Eval()
97 TfLiteEvalTensor* output = in Eval()
A Dfloor_div.cc63 const TfLiteEvalTensor* input1, in EvalFloorDiv()
64 const TfLiteEvalTensor* input2, in EvalFloorDiv()
65 TfLiteEvalTensor* output) { in EvalFloorDiv()
98 const TfLiteEvalTensor* input1 = in Eval()
100 const TfLiteEvalTensor* input2 = in Eval()
102 TfLiteEvalTensor* output = in Eval()
A Dkernel_util.cc23 bool HaveSameShapes(const TfLiteEvalTensor* input1, in HaveSameShapes()
24 const TfLiteEvalTensor* input2) { in HaveSameShapes()
30 const RuntimeShape GetTensorShape(const TfLiteEvalTensor* tensor) { in GetTensorShape()
58 TfLiteEvalTensor* eval_tensor) { in CreateWritableTensorDimsWithCopy()
A Dexpand_dims.cc30 const TfLiteEvalTensor* input, int32_t axis, in ExpandTensorDim()
31 TfLiteEvalTensor* output) { in ExpandTensorDim()
53 const TfLiteEvalTensor* axis, in GetAxisValueFromTensor()
101 const TfLiteEvalTensor* input = in Eval()
103 const TfLiteEvalTensor* axis = in Eval()
105 TfLiteEvalTensor* output = in Eval()
A Ddetection_postprocess.cc264 T ReInterpretTensor(TfLiteEvalTensor* tensor) { in ReInterpretTensor()
272 const TfLiteEvalTensor* input_box_encodings = in DecodeCenterSizeBoxes()
277 const TfLiteEvalTensor* input_anchors = in DecodeCenterSizeBoxes()
407 const TfLiteEvalTensor* input_box_encodings = in NonMaxSuppressionSingleClassHelper()
488 const TfLiteEvalTensor* input_box_encodings = in NonMaxSuppressionMultiClassRegularHelper()
492 TfLiteEvalTensor* detection_boxes = in NonMaxSuppressionMultiClassRegularHelper()
496 TfLiteEvalTensor* detection_scores = in NonMaxSuppressionMultiClassRegularHelper()
498 TfLiteEvalTensor* num_detections = in NonMaxSuppressionMultiClassRegularHelper()
625 TfLiteEvalTensor* detection_boxes = in NonMaxSuppressionMultiClassFastHelper()
630 TfLiteEvalTensor* detection_scores = in NonMaxSuppressionMultiClassFastHelper()
[all …]
A Dsub.cc123 const OpData* data, const TfLiteEvalTensor* input1, in EvalSub()
124 const TfLiteEvalTensor* input2, TfLiteEvalTensor* output) { in EvalSub()
151 const TfLiteEvalTensor* input1, in EvalSubQuantized()
152 const TfLiteEvalTensor* input2, in EvalSubQuantized()
153 TfLiteEvalTensor* output) { in EvalSubQuantized()
217 const TfLiteEvalTensor* input1 = in Eval()
219 const TfLiteEvalTensor* input2 = in Eval()
221 TfLiteEvalTensor* output = in Eval()
A Dsplit.cc29 const TfLiteEvalTensor* input, int axis_value) { in SplitImpl()
32 const TfLiteEvalTensor* output0 = in SplitImpl()
58 TfLiteEvalTensor* t = tflite::micro::GetEvalOutput(context, node, i); in SplitImpl()
83 const TfLiteEvalTensor* axis = tflite::micro::GetEvalInput(context, node, 0); in Eval()
84 const TfLiteEvalTensor* input = tflite::micro::GetEvalInput(context, node, 1); in Eval()
A Dsplit_v.cc30 const TfLiteEvalTensor* input, int axis_value) { in SplitImpl()
32 const TfLiteEvalTensor* output0 = in SplitImpl()
60 TfLiteEvalTensor* output_tensor = in SplitImpl()
88 const TfLiteEvalTensor* input = tflite::micro::GetEvalInput(context, node, 0); in Eval()
89 const TfLiteEvalTensor* axis = tflite::micro::GetEvalInput(context, node, 2); in Eval()
A Dshape.cc31 void ExtractShape(const TfLiteEvalTensor* input, int32_t* output_data) { in ExtractShape()
45 const TfLiteEvalTensor* input = in Eval()
47 TfLiteEvalTensor* output = in Eval()
A Dl2_pool_2d.cc76 TfLiteEvalTensor* output_eval = in L2Prepare()
88 void L2EvalFloat(const TfLitePoolParams& params, const TfLiteEvalTensor& input, in L2EvalFloat()
89 tflite::PoolParams* op_params, TfLiteEvalTensor* output) { in L2EvalFloat()
104 TfLiteEvalTensor* output = in L2Eval()
106 const TfLiteEvalTensor* input = in L2Eval()
A Dkernel_runner.cc85 TfLiteEvalTensor* KernelRunner::GetEvalTensor( in GetEvalTensor()
91 TfLiteEvalTensor* eval_tensor = in GetEvalTensor()
92 reinterpret_cast<TfLiteEvalTensor*>(runner->allocator_->AllocateTemp( in GetEvalTensor()
93 sizeof(TfLiteEvalTensor), alignof(TfLiteEvalTensor))); in GetEvalTensor()
A Dsvdf_common.cc52 const TfLiteEvalTensor* input_tensor, in EvalIntegerSvdfReference()
53 const TfLiteEvalTensor* weights_feature_tensor, in EvalIntegerSvdfReference()
54 const TfLiteEvalTensor* weights_time_tensor, in EvalIntegerSvdfReference()
55 const TfLiteEvalTensor* bias_tensor, in EvalIntegerSvdfReference()
57 TfLiteEvalTensor* activation_state_tensor, in EvalIntegerSvdfReference()
58 TfLiteEvalTensor* output_tensor, in EvalIntegerSvdfReference()
258 TfLiteContext* context, TfLiteNode* node, const TfLiteEvalTensor* input, in EvalFloatSvdfReference()
259 const TfLiteEvalTensor* weights_feature, in EvalFloatSvdfReference()
260 const TfLiteEvalTensor* weights_time, const TfLiteEvalTensor* bias, in EvalFloatSvdfReference()
262 TfLiteEvalTensor* activation_state, TfLiteEvalTensor* output) { in EvalFloatSvdfReference()
A Dunpack.cc32 const TfLiteEvalTensor* input, int output_count, in UnpackImpl()
34 const TfLiteEvalTensor* output0 = in UnpackImpl()
63 TfLiteEvalTensor* t = tflite::micro::GetEvalOutput(context, node, i); in UnpackImpl()
80 const TfLiteEvalTensor* input = in Eval()
A Dbatch_to_space_nd.cc58 const TfLiteEvalTensor* input = in Eval()
60 const TfLiteEvalTensor* block_shape = in Eval()
62 const TfLiteEvalTensor* crops = in Eval()
64 TfLiteEvalTensor* output = in Eval()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/cmsis_nn/
A Dfully_connected.cc105 const TfLiteEvalTensor* input, in EvalQuantizedInt8()
107 const TfLiteEvalTensor* bias, in EvalQuantizedInt8()
108 TfLiteEvalTensor* output) { in EvalQuantizedInt8()
179 const TfLiteEvalTensor* input = in Eval()
181 const TfLiteEvalTensor* filter = in Eval()
183 const TfLiteEvalTensor* bias = in Eval()
185 TfLiteEvalTensor* output = in Eval()
226 const TfLiteEvalTensor* input = in EvalInt8()
228 const TfLiteEvalTensor* filter = in EvalInt8()
230 const TfLiteEvalTensor* bias = in EvalInt8()
[all …]
A Dsvdf.cc135 const TfLiteEvalTensor* weights_feature, in EvalFloatSVDF()
136 const TfLiteEvalTensor* weights_time, const TfLiteEvalTensor* bias, in EvalFloatSVDF()
138 TfLiteEvalTensor* activation_state, TfLiteEvalTensor* output) { in EvalFloatSVDF()
207 const TfLiteEvalTensor* input_tensor, in EvalIntegerSVDF()
210 const TfLiteEvalTensor* bias_tensor, in EvalIntegerSVDF()
212 TfLiteEvalTensor* activation_state_tensor, in EvalIntegerSVDF()
428 const TfLiteEvalTensor* input = in Eval()
430 const TfLiteEvalTensor* weights_feature = in Eval()
432 const TfLiteEvalTensor* weights_time = in Eval()
434 const TfLiteEvalTensor* bias = in Eval()
[all …]
A Dconv.cc133 const OpData& data, const TfLiteEvalTensor* input, in EvalQuantizedPerChannel()
134 const TfLiteEvalTensor* filter, const TfLiteEvalTensor* bias, in EvalQuantizedPerChannel()
135 TfLiteEvalTensor* output, TfLiteEvalTensor* im2col) { in EvalQuantizedPerChannel()
247 const TfLiteEvalTensor* input, in EvalQuantized()
248 const TfLiteEvalTensor* filter, const TfLiteEvalTensor* bias, in EvalQuantized()
249 TfLiteEvalTensor* im2col, TfLiteEvalTensor* hwcn_weights, in EvalQuantized()
250 TfLiteEvalTensor* output) { in EvalQuantized()
284 const TfLiteEvalTensor* input = in Eval()
286 const TfLiteEvalTensor* filter = in Eval()
288 const TfLiteEvalTensor* bias = in Eval()
[all …]
A Dmul.cc105 const TfLiteEvalTensor* input1, in EvalQuantized()
106 const TfLiteEvalTensor* input2, TfLiteEvalTensor* output) { in EvalQuantized()
155 const TfLiteEvalTensor* input1, const TfLiteEvalTensor* input2, in EvalFloat()
156 TfLiteEvalTensor* output) { in EvalFloat()
184 const TfLiteEvalTensor* input1 = in Eval()
186 const TfLiteEvalTensor* input2 = in Eval()
188 TfLiteEvalTensor* output = in Eval()
A Dpooling.cc81 const TfLiteEvalTensor* input, TfLiteEvalTensor* output) { in AverageEvalFloat()
103 const TfLiteEvalTensor* input, in AverageEvalQuantized()
104 TfLiteEvalTensor* output) { in AverageEvalQuantized()
175 const TfLiteEvalTensor* input, TfLiteEvalTensor* output) { in MaxEvalFloat()
197 TfLiteEvalTensor* output) { in MaxEvalQuantizedUInt8()
215 const TfLiteEvalTensor* input, in MaxEvalInt8()
216 TfLiteEvalTensor* output) { in MaxEvalInt8()
330 const TfLiteEvalTensor* input = in AverageEval()
332 TfLiteEvalTensor* output = in AverageEval()
359 const TfLiteEvalTensor* input = in MaxEval()
[all …]
A Dadd.cc108 const OpData* data, const TfLiteEvalTensor* input1, in EvalAdd()
109 const TfLiteEvalTensor* input2, TfLiteEvalTensor* output) { in EvalAdd()
130 const TfLiteEvalTensor* input1, in EvalAddQuantized()
131 const TfLiteEvalTensor* input2, in EvalAddQuantized()
132 TfLiteEvalTensor* output) { in EvalAddQuantized()
216 const TfLiteEvalTensor* input1 = in Eval()
218 const TfLiteEvalTensor* input2 = in Eval()
220 TfLiteEvalTensor* output = in Eval()
A Ddepthwise_conv.cc155 const OpData& data, const TfLiteEvalTensor* input, in EvalQuantizedPerChannel()
156 const TfLiteEvalTensor* filter, in EvalQuantizedPerChannel()
157 const TfLiteEvalTensor* bias, in EvalQuantizedPerChannel()
158 TfLiteEvalTensor* output) { in EvalQuantizedPerChannel()
255 const TfLiteEvalTensor* input, in EvalQuantized()
256 const TfLiteEvalTensor* filter, const TfLiteEvalTensor* bias, in EvalQuantized()
257 TfLiteEvalTensor* output) { in EvalQuantized()
300 TfLiteEvalTensor* output = in Eval()
302 const TfLiteEvalTensor* input = in Eval()
304 const TfLiteEvalTensor* filter = in Eval()
[all …]
A Dsoftmax.cc31 void SoftmaxQuantized(const TfLiteEvalTensor* input, TfLiteEvalTensor* output, in SoftmaxQuantized()
70 const TfLiteEvalTensor* input = tflite::micro::GetEvalInput(context, node, 0); in SoftmaxEval()
71 TfLiteEvalTensor* output = tflite::micro::GetEvalOutput(context, node, 0); in SoftmaxEval()

Completed in 25 milliseconds

123