| /external/tensorflow/tensorflow/lite/kernels/ |
| D | complex_support.cc | 29 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()
|
| D | add_n.cc | 29 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()
|
| D | round.cc | 30 constexpr int kOutputTensor = 0; variable
37 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
51 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
|
| D | zeros_like.cc | 30 constexpr int kOutputTensor = 0; variable
39 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
51 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
|
| D | ceil.cc | 29 constexpr int kOutputTensor = 0; variable
36 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
50 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
|
| D | hashtable_size.cc | 28 constexpr int kOutputTensor = 0; variable
43 context, GetOutputSafe(context, node, kOutputTensor, &output_tensor)); in PrepareHashtableSize()
58 context, GetOutputSafe(context, node, kOutputTensor, &output_tensor)); in EvalHashtableSize()
|
| D | neg.cc | 32 constexpr int kOutputTensor = 0; variable
41 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
53 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
|
| D | floor.cc | 30 constexpr int kOutputTensor = 0; variable
42 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
57 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
|
| D | reverse_sequence.cc | 33 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()
|
| D | where.cc | 30 constexpr int kOutputTensor = 0; variable
64 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
92 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
|
| D | cumsum.cc | 31 static const int kOutputTensor = 0; variable
49 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Prepare()
59 TfLiteTensor* output = GetOutput(context, node, kOutputTensor); in Eval()
|
| D | reshape.cc | 32 constexpr int kOutputTensor = 0; variable
45 GetOutputSafe(context, node, kOutputTensor, &output)); in ResizeOutput()
148 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
165 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
|
| D | comparisons.cc | 37 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()
|
| D | local_response_norm.cc | 36 constexpr int kOutputTensor = 0; variable
46 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
71 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
|
| D | segment_sum.cc | 31 static const int kOutputTensor = 0; variable
75 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
97 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
|
| D | matrix_diag.cc | 30 constexpr int kOutputTensor = 0; variable
43 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
124 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
|
| D | table.cc | 27 constexpr int kOutputTensor = 0; variable
39 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
105 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
|
| D | logical.cc | 33 constexpr int kOutputTensor = 0; variable
65 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
101 GetOutputSafe(context, node, kOutputTensor, &output)); in LogicalImpl()
|
| D | rank.cc | 28 constexpr int kOutputTensor = 0; variable
38 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
|
| D | floor_div.cc | 37 constexpr int kOutputTensor = 0; variable
69 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
138 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
|
| D | pow.cc | 34 constexpr int kOutputTensor = 0; variable
65 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
129 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
|
| D | reverse.cc | 32 constexpr int kOutputTensor = 0; variable
67 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
89 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
|
| D | space_to_depth.cc | 39 constexpr int kOutputTensor = 0; variable
52 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
91 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
|
| /external/tensorflow/tensorflow/lite/python/testdata/ |
| D | test_registerer.cc | 34 constexpr int kOutputTensor = 0; variable
44 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
57 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
| D | tensorrt_test.cc | 48 const char* kOutputTensor = "output"; variable
70 output->setName(kOutputTensor); in CreateSerializedEngine()
94 const int output_index = engine.getBindingIndex(kOutputTensor); in Execute()
|