Home
last modified time | relevance | path

Searched refs:GetOutputSafe (Results 1 – 25 of 109) sorted by relevance

12345

/external/tensorflow/tensorflow/lite/kernels/
Dnon_max_suppression.cc135 GetOutputSafe(context, node, kSoftNMSOutputTensorSelectedIndices, in Prepare()
139 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, in Prepare()
146 GetOutputSafe(context, node, kSoftNMSOutputTensorNumSelectedIndices, in Prepare()
162 context, GetOutputSafe(context, node, kNMSOutputTensorSelectedIndices, in Prepare()
166 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, in Prepare()
246 GetOutputSafe(context, node, kSoftNMSOutputTensorSelectedIndices, in Eval()
248 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, in Eval()
253 GetOutputSafe(context, node, kSoftNMSOutputTensorNumSelectedIndices, in Eval()
269 context, GetOutputSafe(context, node, kNMSOutputTensorSelectedIndices, in Eval()
271 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, in Eval()
Dunique.cc50 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, kOutputUniqueTensor, in Prepare()
53 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, kOutputIndexTensor, in Prepare()
77 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 1, &output_indexes)); in EvalImpl()
96 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &unique_output)); in EvalImpl()
139 GetOutputSafe(context, node, 1, &output_index_tensor)); in Eval()
Dtopk_v2.cc66 context, GetOutputSafe(context, node, kOutputIndexes, &output_indexes)); in ResizeOutput()
69 context, GetOutputSafe(context, node, kOutputValues, &output_values)); in ResizeOutput()
208 context, GetOutputSafe(context, node, kOutputValues, &output_values)); in Prepare()
221 context, GetOutputSafe(context, node, kOutputIndexes, &output_indexes)); in Prepare()
224 context, GetOutputSafe(context, node, kOutputValues, &output_values)); in Prepare()
234 context, GetOutputSafe(context, node, kOutputValues, &output_values)); in Eval()
237 context, GetOutputSafe(context, node, kOutputIndexes, &output_indexes)); in Eval()
Dhashtable_lookup.cc78 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 1, &hits)); in Prepare()
84 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in Prepare()
104 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in Eval()
106 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 1, &hits)); in Eval()
Ddetection_postprocess.cc173 GetOutputSafe(context, node, kOutputTensorDetectionBoxes, in Prepare()
182 GetOutputSafe(context, node, kOutputTensorDetectionClasses, in Prepare()
190 GetOutputSafe(context, node, kOutputTensorDetectionScores, in Prepare()
198 GetOutputSafe(context, node, kOutputTensorNumDetections, in Prepare()
518 GetOutputSafe(context, node, kOutputTensorDetectionBoxes, in NonMaxSuppressionMultiClassRegularHelper()
522 GetOutputSafe(context, node, kOutputTensorDetectionClasses, in NonMaxSuppressionMultiClassRegularHelper()
526 GetOutputSafe(context, node, kOutputTensorDetectionScores, in NonMaxSuppressionMultiClassRegularHelper()
530 GetOutputSafe(context, node, kOutputTensorNumDetections, in NonMaxSuppressionMultiClassRegularHelper()
660 GetOutputSafe(context, node, kOutputTensorDetectionBoxes, in NonMaxSuppressionMultiClassFastHelper()
664 GetOutputSafe(context, node, kOutputTensorDetectionClasses, in NonMaxSuppressionMultiClassFastHelper()
[all …]
Dcomparisons.cc52 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()
Dround.cc37 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
51 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dif.cc120 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, i, &output)); in Prepare()
179 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, i, &output)); in Eval()
189 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, i, &output)); in Eval()
202 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, i, &output)); in Eval()
Dzeros_like.cc39 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
51 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dceil.cc36 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
50 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dsplit.cc45 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, i, &tensor)); in UseDynamicOutputTensors()
71 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, i, &output)); in ResizeOutputTensors()
92 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, i, &tensor)); in Prepare()
Dreshape.cc45 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.cc64 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
98 GetOutputSafe(context, node, kOutputTensor, &output)); in ReverseSequenceImpl()
134 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dactivations.cc258 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in GenericPrepare()
280 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in ReluPrepare()
314 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in HardSwishPrepare()
355 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in LeakyReluPrepare()
392 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in TanhPrepare()
487 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in SigmoidPrepare()
590 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in SoftmaxPrepare()
661 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in LogSoftmaxPrepare()
690 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in PreluPrepare()
746 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in ReluEval()
[all …]
Dneg.cc41 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
53 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dfloor.cc42 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
57 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dread_variable.cc45 GetOutputSafe(context, node, kOutputValue, &output)); in Prepare()
65 GetOutputSafe(context, node, kOutputValue, &output)); in Eval()
Dwhere.cc64 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
92 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
/external/tensorflow/tensorflow/lite/experimental/kernels/
Dctc_beam_search_decoder.cc85 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, i, &indices)); in Prepare()
89 GetOutputSafe(context, node, i + top_paths, &values)); in Prepare()
92 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, i + 2 * top_paths, in Prepare()
99 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, top_paths * 3, in Prepare()
140 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, p, &indices)); in StoreAllDecodedSequences()
145 GetOutputSafe(context, node, p + top_paths, &values)); in StoreAllDecodedSequences()
149 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, p + 2 * top_paths, in StoreAllDecodedSequences()
229 context, GetOutputSafe(context, node, 3 * top_paths, &log_probabilities)); in Eval()
/external/tensorflow/tensorflow/lite/micro/kernels/
Delu.cc93 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in GenericPrepare()
103 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in EluPrepare()
119 TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output)); in EluEval()
Dfill.cc54 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
71 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
Dadd_n.cc41 GetOutputSafe(context, node, kOutputTensor, &output)); in Prepare()
78 GetOutputSafe(context, node, kOutputTensor, &output)); in Eval()
/external/tensorflow/tensorflow/lite/kernels/hashtable/
Dhashtable_size.cc45 context, GetOutputSafe(context, node, kOutputTensor, &output_tensor)); in PrepareHashtableSize()
60 context, GetOutputSafe(context, node, kOutputTensor, &output_tensor)); in EvalHashtableSize()
/external/tensorflow/tensorflow/lite/java/src/test/native/
Dinterpreter_test_jni.cc39 tflite::GetOutputSafe(context, node, 0, &output)); in Java_org_tensorflow_lite_InterpreterTest_getNativeHandleForDelegate()
48 tflite::GetOutputSafe(context, node, 0, &output)); in Java_org_tensorflow_lite_InterpreterTest_getNativeHandleForDelegate()
/external/tensorflow/tensorflow/lite/kernels/perception/
Dmax_pool_with_argmax.cc161 GetOutputSafe(context, node, kDataOutputTensor, &output)); in Prepare()
163 context, GetOutputSafe(context, node, kIndicesOutputTensor, &indices)); in Prepare()
218 GetOutputSafe(context, node, kDataOutputTensor, &output)); in Eval()
220 context, GetOutputSafe(context, node, kIndicesOutputTensor, &indices)); in Eval()

12345