Home
last modified time | relevance | path

Searched refs:output_selected_scores (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dnon_max_suppression.cc138 TfLiteTensor* output_selected_scores; in Prepare() local
141 &output_selected_scores)); in Prepare()
142 output_selected_scores->type = kTfLiteFloat32; in Prepare()
153 SetTensorSizes(context, output_selected_scores, {max_output_size_value}); in Prepare()
156 SetTensorToDynamic(output_selected_scores); in Prepare()
230 TfLiteTensor* output_selected_scores = nullptr; in Eval() local
250 &output_selected_scores)); in Eval()
257 SetTensorSizes(context, output_selected_scores, {max_output_size_value}); in Eval()
262 output_selected_indices->data.i32, output_selected_scores->data.f, in Eval()
266 output_selected_indices->data.i32, output_selected_scores->data.f); in Eval()