Searched refs:SetTensorSizes (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | non_max_suppression.cc | 62 TfLiteStatus SetTensorSizes(TfLiteContext* context, TfLiteTensor* tensor, in SetTensorSizes() function
149 SetTensorSizes(context, output_num_selected_indices, {}); in Prepare()
152 SetTensorSizes(context, output_selected_indices, {max_output_size_value}); in Prepare()
153 SetTensorSizes(context, output_selected_scores, {max_output_size_value}); in Prepare()
170 SetTensorSizes(context, output_num_selected_indices, {}); in Prepare()
173 SetTensorSizes(context, output_selected_indices, {max_output_size_value}); in Prepare()
256 SetTensorSizes(context, output_selected_indices, {max_output_size_value}); in Eval()
257 SetTensorSizes(context, output_selected_scores, {max_output_size_value}); in Eval()
275 SetTensorSizes(context, output_selected_indices, {max_output_size_value}); in Eval()
|
| D | detection_postprocess.cc | 134 TfLiteStatus SetTensorSizes(TfLiteContext* context, TfLiteTensor* tensor, in SetTensorSizes() function
176 SetTensorSizes(context, detection_boxes, in Prepare()
185 SetTensorSizes(context, detection_classes, {kBatchSize, num_detected_boxes}); in Prepare()
193 SetTensorSizes(context, detection_scores, {kBatchSize, num_detected_boxes}); in Prepare()
201 SetTensorSizes(context, num_detections, {1}); in Prepare()
214 SetTensorSizes(context, decoded_boxes, in Prepare()
221 SetTensorSizes(context, scores, in Prepare()
230 SetTensorSizes(context, active_candidate, in Prepare()
|