Searched refs:SetTensorSizes (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | detection_postprocess.cc | 129 TfLiteStatus SetTensorSizes(TfLiteContext* context, TfLiteTensor* tensor, in SetTensorSizes() function
164 SetTensorSizes(context, detection_boxes, in Prepare()
171 SetTensorSizes(context, detection_classes, {kBatchSize, num_detected_boxes}); in Prepare()
177 SetTensorSizes(context, detection_scores, {kBatchSize, num_detected_boxes}); in Prepare()
184 SetTensorSizes(context, num_detections, {1}); in Prepare()
197 SetTensorSizes(context, decoded_boxes, in Prepare()
204 SetTensorSizes(context, scores, in Prepare()
213 SetTensorSizes(context, active_candidate, in Prepare()
|