Home
last modified time | relevance | path

Searched refs:kOutputTensor (Results 1 – 25 of 128) sorted by relevance

123456

/external/tensorflow/tensorflow/lite/kernels/
Dcomplex_support.cc29 static const int kOutputTensor = 0; variable
40 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
66 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in EvalReal()
98 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in EvalImag()
129 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in EvalAbs()
Dadd_n.cc29 constexpr int kOutputTensor = 0; variable
41 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
64 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in EvalAddN()
78 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dround.cc30 constexpr int kOutputTensor = 0; variable
37 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
51 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dceil.cc29 constexpr int kOutputTensor = 0; variable
36 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
50 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dzeros_like.cc30 constexpr int kOutputTensor = 0; variable
39 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
51 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dneg.cc32 constexpr int kOutputTensor = 0; variable
41 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
53 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dfloor.cc30 constexpr int kOutputTensor = 0; variable
42 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
57 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dwhere.cc30 constexpr int kOutputTensor = 0; variable
64 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
92 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dreshape.cc32 constexpr int kOutputTensor = 0; variable
45 GetOutputSafe(context, node, kOutputTensor, &output)); in ResizeOutput()
132 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
149 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dreverse_sequence.cc33 constexpr int kOutputTensor = 0; variable
64 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
98 GetOutputSafe(context, node, kOutputTensor, &output)); in ReverseSequenceImpl()
134 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dcumsum.cc31 static const int kOutputTensor = 0; variable
49 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
59 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Eval()
Dlocal_response_norm.cc36 constexpr int kOutputTensor = 0; variable
46 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
71 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dcomparisons.cc37 constexpr int kOutputTensor = 0; variable
52 GetOutputSafe(context, node, kOutputTensor, &output)); in ComparisonPrepareCommon()
162 GetOutputSafe(context, node, kOutputTensor, &output)); in EqualEval()
212 GetOutputSafe(context, node, kOutputTensor, &output)); in NotEqualEval()
262 GetOutputSafe(context, node, kOutputTensor, &output)); in GreaterEval()
303 GetOutputSafe(context, node, kOutputTensor, &output)); in GreaterEqualEval()
344 GetOutputSafe(context, node, kOutputTensor, &output)); in LessEval()
385 GetOutputSafe(context, node, kOutputTensor, &output)); in LessEqualEval()
/external/tensorflow/tensorflow/lite/micro/kernels/
Dadd_n.cc29 constexpr int kOutputTensor = 0; variable
41 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
64 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in EvalAddN()
78 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dzeros_like.cc25 constexpr int kOutputTensor = 0; variable
34 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
51 tflite::micro::GetEvalOutput(context, node, kOutputTensor); in Eval()
Dceil.cc29 constexpr int kOutputTensor = 0; variable
34 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
52 tflite::micro::GetEvalOutput(context, node, kOutputTensor); in Eval()
Dround.cc29 constexpr int kOutputTensor = 0; variable
34 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
52 tflite::micro::GetEvalOutput(context, node, kOutputTensor); in Eval()
Dconcatenation.cc33 constexpr int kOutputTensor = 0; variable
97 tflite::micro::GetEvalOutput(context, node, kOutputTensor); in EvalUnquantized()
117 tflite::micro::GetEvalOutput(context, node, kOutputTensor); in EvalQuantizedUInt8()
142 const TfLiteTensor* output_tensor = GetOutput(context, node, kOutputTensor); in Prepare()
180 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
230 const TfLiteTensor* output_tensor = GetOutput(context, node, kOutputTensor); in Eval()
Dexp.cc27 constexpr int kOutputTensor = 0; variable
34 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
50 tflite::micro::GetEvalOutput(context, node, kOutputTensor); in Eval()
Dcast.cc25 constexpr int kOutputTensor = 0; variable
32 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
66 tflite::micro::GetEvalOutput(context, node, kOutputTensor); in Eval()
Dfill.cc31 constexpr int kOutputTensor = 0; variable
54 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
71 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dreshape.cc31 constexpr int kOutputTensor = 0; variable
36 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in ReshapeOutput()
85 tflite::micro::GetEvalOutput(context, node, kOutputTensor); in Eval()
Dbatch_to_space_nd.cc31 constexpr int kOutputTensor = 0; variable
45 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
69 tflite::micro::GetEvalOutput(context, node, kOutputTensor); in Eval()
Dfloor.cc28 constexpr int kOutputTensor = 0; variable
35 tflite::micro::GetEvalOutput(context, node, kOutputTensor); in Eval()
/external/tensorflow/tensorflow/lite/kernels/hashtable/
Dhashtable_size.cc29 constexpr int kOutputTensor = 0; variable
45 context, GetOutputSafe(context, node, kOutputTensor, &output_tensor)); in PrepareHashtableSize()
60 context, GetOutputSafe(context, node, kOutputTensor, &output_tensor)); in EvalHashtableSize()

123456