Searched refs:num_detected_boxes (Results 1 – 2 of 2) sorted by relevance
| /external/ComputeLibrary/src/runtime/CPP/functions/ |
| D | CPPDetectionPostProcessLayer.cpp | 69 … const unsigned int num_detected_boxes = info.max_detections() * info.max_classes_per_detection(); in validate_arguments() local
74 …R_ON_MISMATCHING_DIMENSIONS(output_boxes->tensor_shape(), TensorShape(4U, num_detected_boxes, 1U)); in validate_arguments()
79 …ROR_ON_MISMATCHING_DIMENSIONS(output_classes->tensor_shape(), TensorShape(num_detected_boxes, 1U)); in validate_arguments()
84 …RROR_ON_MISMATCHING_DIMENSIONS(output_scores->tensor_shape(), TensorShape(num_detected_boxes, 1U)); in validate_arguments()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | detection_postprocess.cc | 165 const int num_detected_boxes = in Prepare() local
178 {kBatchSize, num_detected_boxes, kNumCoordBox}); in Prepare()
186 SetTensorSizes(context, detection_classes, {kBatchSize, num_detected_boxes}); in Prepare()
194 SetTensorSizes(context, detection_scores, {kBatchSize, num_detected_boxes}); in Prepare()
|