Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/lite/micro/kernels/
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);
Dsvdf.h40 const TfLiteEvalTensor* input_tensor,
41 const TfLiteEvalTensor* weights_feature_tensor,
42 const TfLiteEvalTensor* weights_time_tensor,
43 const TfLiteEvalTensor* bias_tensor,
45 TfLiteEvalTensor* activation_state_tensor,
46 TfLiteEvalTensor* output_tensor,
Ddepthwise_conv.cc170 const TfLiteEvalTensor* input, const TfLiteEvalTensor* filter, in EvalFloat()
171 const TfLiteEvalTensor* bias, TfLiteEvalTensor* output) { in EvalFloat()
202 const OpData& data, const TfLiteEvalTensor* input, in EvalQuantizedPerChannel()
203 const TfLiteEvalTensor* filter, in EvalQuantizedPerChannel()
204 const TfLiteEvalTensor* bias, in EvalQuantizedPerChannel()
205 TfLiteEvalTensor* output) { in EvalQuantizedPerChannel()
236 const TfLiteEvalTensor* input, in EvalQuantized()
237 const TfLiteEvalTensor* filter, const TfLiteEvalTensor* bias, in EvalQuantized()
238 TfLiteEvalTensor* output) { in EvalQuantized()
281 TfLiteEvalTensor* output = in Eval()
[all …]
Dpooling.cc63 const TfLiteEvalTensor* input, TfLiteEvalTensor* output) { in AverageEvalFloat()
81 const TfLiteEvalTensor* input, in AverageEvalQuantized()
82 TfLiteEvalTensor* output) { in AverageEvalQuantized()
111 const TfLiteEvalTensor* input, TfLiteEvalTensor* output) { in MaxEvalFloat()
129 const TfLiteEvalTensor* input, TfLiteEvalTensor* output) { in MaxEvalQuantized()
162 const TfLiteEvalTensor* input = in AverageEval()
164 TfLiteEvalTensor* output = in AverageEval()
191 const TfLiteEvalTensor* input = in MaxEval()
193 TfLiteEvalTensor* output = in MaxEval()
Ddetection_postprocess.cc238 void DequantizeBoxEncodings(const TfLiteEvalTensor* input_box_encodings, in DequantizeBoxEncodings()
258 T ReInterpretTensor(const TfLiteEvalTensor* tensor) { in ReInterpretTensor()
264 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()
490 const TfLiteEvalTensor* input_class_predictions = in NonMaxSuppressionMultiClassRegularHelper()
492 TfLiteEvalTensor* detection_boxes = in NonMaxSuppressionMultiClassRegularHelper()
494 TfLiteEvalTensor* detection_classes = tflite::micro::GetEvalOutput( in NonMaxSuppressionMultiClassRegularHelper()
[all …]
Dsvdf.cc118 TfLiteContext* context, TfLiteNode* node, const TfLiteEvalTensor* input, in EvalFloatSVDF()
119 const TfLiteEvalTensor* weights_feature, in EvalFloatSVDF()
120 const TfLiteEvalTensor* weights_time, const TfLiteEvalTensor* bias, in EvalFloatSVDF()
122 TfLiteEvalTensor* activation_state, TfLiteEvalTensor* output) { in EvalFloatSVDF()
338 const TfLiteEvalTensor* input = in Eval()
340 const TfLiteEvalTensor* weights_feature = in Eval()
342 const TfLiteEvalTensor* weights_time = in Eval()
344 const TfLiteEvalTensor* bias = in Eval()
348 TfLiteEvalTensor* activation_state = tflite::micro::GetMutableEvalInput( in Eval()
350 TfLiteEvalTensor* output = in Eval()
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()
Dadd.cc108 const OpData* data, const TfLiteEvalTensor* input1, in EvalAdd()
109 const TfLiteEvalTensor* input2, TfLiteEvalTensor* output) { in EvalAdd()
133 const TfLiteEvalTensor* input1, in EvalAddQuantized()
134 const TfLiteEvalTensor* input2, in EvalAddQuantized()
135 TfLiteEvalTensor* output) { in EvalAddQuantized()
225 const TfLiteEvalTensor* input1 = in Eval()
227 const TfLiteEvalTensor* input2 = in Eval()
229 TfLiteEvalTensor* output = in Eval()
Dmul.cc91 const TfLiteEvalTensor* input1, in EvalQuantized()
92 const TfLiteEvalTensor* input2, TfLiteEvalTensor* output) { in EvalQuantized()
148 const TfLiteEvalTensor* input1, const TfLiteEvalTensor* input2, in EvalFloat()
149 TfLiteEvalTensor* output) { in EvalFloat()
198 const TfLiteEvalTensor* input1 = in Eval()
200 const TfLiteEvalTensor* input2 = in Eval()
202 TfLiteEvalTensor* output = in Eval()
Dsvdf_common.cc33 const TfLiteEvalTensor* input_tensor, in EvalIntegerSvdfReference()
34 const TfLiteEvalTensor* weights_feature_tensor, in EvalIntegerSvdfReference()
35 const TfLiteEvalTensor* weights_time_tensor, in EvalIntegerSvdfReference()
36 const TfLiteEvalTensor* bias_tensor, in EvalIntegerSvdfReference()
38 TfLiteEvalTensor* activation_state_tensor, in EvalIntegerSvdfReference()
39 TfLiteEvalTensor* output_tensor, in EvalIntegerSvdfReference()
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()
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()
/external/tensorflow/tensorflow/lite/micro/
Dmicro_allocator.h127 TfLiteEvalTensor** eval_tensors);
138 const Model* model, TfLiteEvalTensor* eval_tensors,
148 const Model* model, TfLiteEvalTensor* eval_tensors, int tensor_index);
157 TfLiteEvalTensor* eval_tensors,
210 const Model* model, TfLiteEvalTensor** eval_tensors);
214 TfLiteEvalTensor* eval_tensors);
220 const Model* model, TfLiteEvalTensor* eval_tensors, int tensor_index);
244 TfLiteEvalTensor* eval_tensors,
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Dfully_connected.cc165 const TfLiteEvalTensor* input, in EvalMliQuantizedInt8()
166 const TfLiteEvalTensor* filter, in EvalMliQuantizedInt8()
167 const TfLiteEvalTensor* bias, in EvalMliQuantizedInt8()
168 TfLiteEvalTensor* output) { in EvalMliQuantizedInt8()
259 const TfLiteEvalTensor* input, in EvalQuantizedInt8()
260 const TfLiteEvalTensor* filter, in EvalQuantizedInt8()
261 const TfLiteEvalTensor* bias, in EvalQuantizedInt8()
262 TfLiteEvalTensor* output) { in EvalQuantizedInt8()
291 const OpData& data, const TfLiteEvalTensor* input, in EvalQuantized()
292 const TfLiteEvalTensor* filter, in EvalQuantized()
[all …]
Dconv.cc269 const TfLiteEvalTensor* input, in EvalQuantized()
270 const TfLiteEvalTensor* filter, const TfLiteEvalTensor* bias, in EvalQuantized()
271 TfLiteEvalTensor* im2col, TfLiteEvalTensor* hwcn_weights, in EvalQuantized()
272 TfLiteEvalTensor* output) { in EvalQuantized()
312 const OpData& data, const TfLiteEvalTensor* input, in EvalMliQuantizedPerChannel()
313 const TfLiteEvalTensor* filter, const TfLiteEvalTensor* bias, in EvalMliQuantizedPerChannel()
314 TfLiteEvalTensor* output) { in EvalMliQuantizedPerChannel()
440 const TfLiteEvalTensor* input, in EvalQuantizedPerChannel()
441 const TfLiteEvalTensor* filter, in EvalQuantizedPerChannel()
442 const TfLiteEvalTensor* bias, in EvalQuantizedPerChannel()
[all …]
Ddepthwise_conv.cc262 const TfLiteEvalTensor* input, const TfLiteEvalTensor* filter, in EvalFloat()
263 const TfLiteEvalTensor* bias, TfLiteEvalTensor* output) { in EvalFloat()
299 const OpData& data, const TfLiteEvalTensor* input, in EvalMliQuantizedPerChannel()
300 const TfLiteEvalTensor* filter, const TfLiteEvalTensor* bias, in EvalMliQuantizedPerChannel()
301 TfLiteEvalTensor* output) { in EvalMliQuantizedPerChannel()
450 const OpData& data, const TfLiteEvalTensor* input, in EvalQuantizedPerChannel()
451 const TfLiteEvalTensor* filter, in EvalQuantizedPerChannel()
452 const TfLiteEvalTensor* bias, in EvalQuantizedPerChannel()
453 TfLiteEvalTensor* output) { in EvalQuantizedPerChannel()
488 const TfLiteEvalTensor* input, in EvalQuantized()
[all …]
Dpooling.cc146 const TfLiteEvalTensor* input, TfLiteEvalTensor* output) { in AverageEvalFloat()
174 const OpData& data, const TfLiteEvalTensor* input, in EvalMli()
175 TfLiteEvalTensor* output, in EvalMli()
239 const TfLiteEvalTensor* input, in AverageEvalQuantized()
240 TfLiteEvalTensor* output) { in AverageEvalQuantized()
275 const TfLiteEvalTensor* input, TfLiteEvalTensor* output) { in MaxEvalFloat()
300 const TfLiteEvalTensor* input, TfLiteEvalTensor* output) { in MaxEvalQuantized()
337 const TfLiteEvalTensor* input = in AverageEval()
339 TfLiteEvalTensor* output = in AverageEval()
369 const TfLiteEvalTensor* input = in MaxEval()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dsvdf.cc134 TfLiteContext* context, TfLiteNode* node, const TfLiteEvalTensor* input, in EvalFloatSVDF()
135 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()
208 const TfLiteEvalTensor* weights_feature_tensor, in EvalIntegerSVDF()
209 const TfLiteEvalTensor* weights_time_tensor, in EvalIntegerSVDF()
210 const TfLiteEvalTensor* bias_tensor, in EvalIntegerSVDF()
212 TfLiteEvalTensor* activation_state_tensor, in EvalIntegerSVDF()
213 TfLiteEvalTensor* output_tensor, const OpData& data) { in EvalIntegerSVDF()
[all …]
Dfully_connected.cc101 const TfLiteEvalTensor* input, in EvalQuantizedInt8()
102 const TfLiteEvalTensor* filter, in EvalQuantizedInt8()
103 const TfLiteEvalTensor* bias, in EvalQuantizedInt8()
104 TfLiteEvalTensor* output) { in EvalQuantizedInt8()
189 const TfLiteEvalTensor* input = in Eval()
191 const TfLiteEvalTensor* filter = in Eval()
193 const TfLiteEvalTensor* bias = in Eval()
195 TfLiteEvalTensor* output = in Eval()
249 const TfLiteEvalTensor* input = in EvalInt8()
251 const TfLiteEvalTensor* filter = in EvalInt8()
[all …]
Dmul.cc104 const TfLiteEvalTensor* input1, in EvalQuantized()
105 const TfLiteEvalTensor* input2, TfLiteEvalTensor* output) { in EvalQuantized()
155 TfLiteMulParams* params, const TfLiteEvalTensor* input1, in EvalFloat()
156 const TfLiteEvalTensor* input2, TfLiteEvalTensor* output) { in EvalFloat()
185 const TfLiteEvalTensor* input1 = in Eval()
187 const TfLiteEvalTensor* input2 = in Eval()
189 TfLiteEvalTensor* output = in Eval()
Dpooling.cc76 const TfLiteEvalTensor* input, TfLiteEvalTensor* output) { in AverageEvalFloat()
98 const TfLiteEvalTensor* input, in AverageEvalQuantized()
99 TfLiteEvalTensor* output) { in AverageEvalQuantized()
170 const TfLiteEvalTensor* input, TfLiteEvalTensor* output) { in MaxEvalFloat()
191 const TfLiteEvalTensor* input, in MaxEvalQuantizedUInt8()
192 TfLiteEvalTensor* output) { in MaxEvalQuantizedUInt8()
210 const TfLiteEvalTensor* input, in MaxEvalInt8()
211 TfLiteEvalTensor* output) { in MaxEvalInt8()
321 const TfLiteEvalTensor* input = in AverageEval()
323 TfLiteEvalTensor* output = in AverageEval()
[all …]
Dadd.cc100 const OpData* data, const TfLiteEvalTensor* input1, in EvalAdd()
101 const TfLiteEvalTensor* input2, TfLiteEvalTensor* output) { in EvalAdd()
124 const TfLiteEvalTensor* input1, in EvalAddQuantized()
125 const TfLiteEvalTensor* input2, in EvalAddQuantized()
126 TfLiteEvalTensor* output) { in EvalAddQuantized()
207 const TfLiteEvalTensor* input1 = in Eval()
209 const TfLiteEvalTensor* input2 = in Eval()
211 TfLiteEvalTensor* output = in Eval()
Ddepthwise_conv.cc213 const TfLiteEvalTensor* input, const TfLiteEvalTensor* filter, in EvalFloat()
214 const TfLiteEvalTensor* bias, TfLiteEvalTensor* output) { in EvalFloat()
245 const TfLiteEvalTensor* input, in EvalQuantizedPerChannel()
246 const TfLiteEvalTensor* filter, in EvalQuantizedPerChannel()
247 const TfLiteEvalTensor* bias, in EvalQuantizedPerChannel()
248 TfLiteEvalTensor* output) { in EvalQuantizedPerChannel()
358 const TfLiteEvalTensor* input, in EvalQuantized()
359 const TfLiteEvalTensor* filter, const TfLiteEvalTensor* bias, in EvalQuantized()
360 TfLiteEvalTensor* output) { in EvalQuantized()
430 TfLiteEvalTensor* output = in Eval()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dsvdf.cc55 const TfLiteEvalTensor* input_tensor, in EvalIntegerSvdfHifimini()
56 const TfLiteEvalTensor* weights_feature_tensor, in EvalIntegerSvdfHifimini()
57 const TfLiteEvalTensor* weights_time_tensor, in EvalIntegerSvdfHifimini()
58 const TfLiteEvalTensor* bias_tensor, in EvalIntegerSvdfHifimini()
60 TfLiteEvalTensor* activation_state_tensor, in EvalIntegerSvdfHifimini()
61 TfLiteEvalTensor* output_tensor, OpData data) { in EvalIntegerSvdfHifimini()
251 const TfLiteEvalTensor* input_tensor, in EvalIntegerSvdfHifi4()
252 const TfLiteEvalTensor* weights_feature_tensor, in EvalIntegerSvdfHifi4()
253 const TfLiteEvalTensor* weights_time_tensor, in EvalIntegerSvdfHifi4()
254 const TfLiteEvalTensor* bias_tensor, const TfLiteSVDFParams* params, in EvalIntegerSvdfHifi4()
[all …]
Dfully_connected.cc184 const TfLiteEvalTensor* input, in EvalQuantizedInt8()
185 const TfLiteEvalTensor* filter, in EvalQuantizedInt8()
186 const TfLiteEvalTensor* bias, in EvalQuantizedInt8()
187 TfLiteEvalTensor* output) { in EvalQuantizedInt8()
255 const TfLiteEvalTensor* input = in Eval()
257 const TfLiteEvalTensor* filter = in Eval()
259 const TfLiteEvalTensor* bias = in Eval()
264 TfLiteEvalTensor* output = in Eval()

1234