Home
last modified time | relevance | path

Searched refs:GetEvalInput (Results 1 – 25 of 65) sorted by relevance

123

/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dfully_connected.cc190 tflite::micro::GetEvalInput(context, node, kFullyConnectedInputTensor); in Eval()
192 tflite::micro::GetEvalInput(context, node, kFullyConnectedWeightsTensor); in Eval()
194 tflite::micro::GetEvalInput(context, node, kFullyConnectedBiasTensor); in Eval()
250 tflite::micro::GetEvalInput(context, node, kFullyConnectedInputTensor); in EvalInt8()
252 tflite::micro::GetEvalInput(context, node, kFullyConnectedWeightsTensor); in EvalInt8()
254 tflite::micro::GetEvalInput(context, node, kFullyConnectedBiasTensor); in EvalInt8()
Dsvdf.cc425 tflite::micro::GetEvalInput(context, node, kInputTensor); in Eval()
427 tflite::micro::GetEvalInput(context, node, kWeightsFeatureTensor); in Eval()
429 tflite::micro::GetEvalInput(context, node, kWeightsTimeTensor); in Eval()
432 ? tflite::micro::GetEvalInput(context, node, kBiasTensor) in Eval()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dbatch_to_space_nd.cc63 tflite::micro::GetEvalInput(context, node, kInputTensor); in Eval()
65 tflite::micro::GetEvalInput(context, node, kBlockShapeTensor); in Eval()
67 tflite::micro::GetEvalInput(context, node, kCropsTensor); in Eval()
Dconv.cc39 tflite::micro::GetEvalInput(context, node, kConvInputTensor); in Eval()
41 tflite::micro::GetEvalInput(context, node, kConvWeightsTensor); in Eval()
44 ? tflite::micro::GetEvalInput(context, node, kConvBiasTensor) in Eval()
Dspace_to_batch_nd.cc73 tflite::micro::GetEvalInput(context, node, kInputTensor); in Eval()
75 tflite::micro::GetEvalInput(context, node, kBlockShapeTensor); in Eval()
77 tflite::micro::GetEvalInput(context, node, kCropsTensor); in Eval()
Dfully_connected.cc70 tflite::micro::GetEvalInput(context, node, kFullyConnectedInputTensor); in Eval()
72 tflite::micro::GetEvalInput(context, node, kFullyConnectedWeightsTensor); in Eval()
74 tflite::micro::GetEvalInput(context, node, kFullyConnectedBiasTensor); in Eval()
Dcomparisons.cc42 tflite::micro::GetEvalInput(context, node, kInputTensor1); in EqualEval()
44 tflite::micro::GetEvalInput(context, node, kInputTensor2); in EqualEval()
147 tflite::micro::GetEvalInput(context, node, kInputTensor1); in NotEqualEval()
149 tflite::micro::GetEvalInput(context, node, kInputTensor2); in NotEqualEval()
251 tflite::micro::GetEvalInput(context, node, kInputTensor1); in GreaterEval()
253 tflite::micro::GetEvalInput(context, node, kInputTensor2); in GreaterEval()
342 tflite::micro::GetEvalInput(context, node, kInputTensor1); in GreaterEqualEval()
344 tflite::micro::GetEvalInput(context, node, kInputTensor2); in GreaterEqualEval()
433 tflite::micro::GetEvalInput(context, node, kInputTensor1); in LessEval()
435 tflite::micro::GetEvalInput(context, node, kInputTensor2); in LessEval()
[all …]
Ddetection_postprocess.cc273 tflite::micro::GetEvalInput(context, node, kInputTensorBoxEncodings); in DecodeCenterSizeBoxes()
278 tflite::micro::GetEvalInput(context, node, kInputTensorAnchors); in DecodeCenterSizeBoxes()
408 tflite::micro::GetEvalInput(context, node, kInputTensorBoxEncodings); in NonMaxSuppressionSingleClassHelper()
489 tflite::micro::GetEvalInput(context, node, kInputTensorBoxEncodings); in NonMaxSuppressionMultiClassRegularHelper()
491 tflite::micro::GetEvalInput(context, node, kInputTensorClassPredictions); in NonMaxSuppressionMultiClassRegularHelper()
622 tflite::micro::GetEvalInput(context, node, kInputTensorBoxEncodings); in NonMaxSuppressionMultiClassFastHelper()
624 tflite::micro::GetEvalInput(context, node, kInputTensorClassPredictions); in NonMaxSuppressionMultiClassFastHelper()
724 tflite::micro::GetEvalInput(context, node, kInputTensorBoxEncodings); in NonMaxSuppressionMultiClass()
726 tflite::micro::GetEvalInput(context, node, kInputTensorClassPredictions); in NonMaxSuppressionMultiClass()
Dlogical.cc35 tflite::micro::GetEvalInput(context, node, kInputTensor1); in LogicalImpl()
37 tflite::micro::GetEvalInput(context, node, kInputTensor2); in LogicalImpl()
Darg_min_max.cc50 tflite::micro::GetEvalInput(context, node, kInputTensor); in Eval()
52 tflite::micro::GetEvalInput(context, node, kAxis); in Eval()
Dpack.cc33 tflite::micro::GetEvalInput(context, node, 0); in PackImpl()
60 const TfLiteEvalTensor* t = tflite::micro::GetEvalInput(context, node, i); in PackImpl()
Dmaximum_minimum.cc44 input1 = tflite::micro::GetEvalInput(context, node, kInputTensor1); in OpContext()
45 input2 = tflite::micro::GetEvalInput(context, node, kInputTensor2); in OpContext()
Dresize_nearest_neighbor.cc63 tflite::micro::GetEvalInput(context, node, kInputTensor); in Eval()
65 tflite::micro::GetEvalInput(context, node, kSizeTensor); in Eval()
Dsplit.cc83 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.cc88 const TfLiteEvalTensor* input = tflite::micro::GetEvalInput(context, node, 0); in Eval()
89 const TfLiteEvalTensor* axis = tflite::micro::GetEvalInput(context, node, 2); in Eval()
Dsvdf.cc339 tflite::micro::GetEvalInput(context, node, kInputTensor); in Eval()
341 tflite::micro::GetEvalInput(context, node, kWeightsFeatureTensor); in Eval()
343 tflite::micro::GetEvalInput(context, node, kWeightsTimeTensor); in Eval()
346 ? tflite::micro::GetEvalInput(context, node, kBiasTensor) in Eval()
Dreduce.cc140 const TfLiteEvalTensor* input = tflite::micro::GetEvalInput(context, node, 0); in EvalMean()
141 const TfLiteEvalTensor* axis = tflite::micro::GetEvalInput(context, node, 1); in EvalMean()
262 const TfLiteEvalTensor* input = tflite::micro::GetEvalInput(context, node, 0); in EvalMax()
263 const TfLiteEvalTensor* axis = tflite::micro::GetEvalInput(context, node, 1); in EvalMax()
Dtranspose_conv.cc201 tflite::micro::GetEvalInput(context, node, kInputTensor); in Eval()
203 tflite::micro::GetEvalInput(context, node, kFilterTensor); in Eval()
206 ? tflite::micro::GetEvalInput(context, node, kBiasTensor) in Eval()
Dprelu.cc112 const TfLiteEvalTensor* input = tflite::micro::GetEvalInput(context, node, 0); in PreluEval()
113 const TfLiteEvalTensor* alpha = tflite::micro::GetEvalInput(context, node, 1); in PreluEval()
Dfloor.cc32 tflite::micro::GetEvalInput(context, node, kInputTensor); in Eval()
Dneg.cc32 tflite::micro::GetEvalInput(context, node, kInputTensor); in Eval()
Dshape.cc46 tflite::micro::GetEvalInput(context, node, kInputTensor); in Eval()
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dfully_connected.cc256 tflite::micro::GetEvalInput(context, node, kFullyConnectedInputTensor); in Eval()
258 tflite::micro::GetEvalInput(context, node, kFullyConnectedWeightsTensor); in Eval()
260 (NumInputs(node) == 3) ? tflite::micro::GetEvalInput( in Eval()
Dconv.cc252 tflite::micro::GetEvalInput(context, node, kConvInputTensor); in Eval()
254 tflite::micro::GetEvalInput(context, node, kConvWeightsTensor); in Eval()
257 ? tflite::micro::GetEvalInput(context, node, kConvBiasTensor) in Eval()
Dsvdf.cc454 tflite::micro::GetEvalInput(context, node, kInputTensor); in Eval()
456 tflite::micro::GetEvalInput(context, node, kWeightsFeatureTensor); in Eval()
458 tflite::micro::GetEvalInput(context, node, kWeightsTimeTensor); in Eval()
461 ? tflite::micro::GetEvalInput(context, node, kBiasTensor) in Eval()

123