| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEDetectionPostProcessLayer.cpp | 43 …tput_boxes, ITensor *output_classes, ITensor *output_scores, ITensor *num_detection, DetectionPost… in configure() argument
48 num_detection->info(), info)); in configure()
49 …ng, input_scores, input_anchors, output_boxes, output_classes, output_scores, num_detection, info); in configure()
70 …es_to_use, input_anchors, output_boxes, output_classes, output_scores, num_detection, info_to_use); in configure()
75 …ITensorInfo *output_classes, ITensorInfo *output_scores, ITensorInfo *num_detection, DetectionPost… in validate() argument
83 …g, input_scores, input_anchors, output_boxes, output_classes, output_scores, num_detection, info)); in validate()
|
| /external/ComputeLibrary/src/runtime/CPP/functions/ |
| D | CPPDetectionPostProcessLayer.cpp | 42 …*output_boxes, ITensorInfo *output_classes, ITensorInfo *output_scores, ITensorInfo *num_detection, in validate_arguments() argument
65 …ARM_COMPUTE_RETURN_ERROR_ON_MSG(num_detection->num_dimensions() > 1, "The num_detection output ten… in validate_arguments()
87 if(num_detection->total_size() != 0) in validate_arguments()
89 …ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DIMENSIONS(num_detection->tensor_shape(), TensorShape(1U)); in validate_arguments()
90 ARM_COMPUTE_RETURN_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(num_detection, 1, DataType::F32); in validate_arguments()
184 ITensor *num_detection) in SaveOutputs() argument
207 *(reinterpret_cast<float *>(num_detection->ptr_to_element(Coordinates(0)))) = num_output; in SaveOutputs()
220 … ITensor *output_scores, ITensor *num_detection, DetectionPostProcessLayerInfo info) in configure() argument
224 num_detection, info); in configure()
231 auto_init_if_empty(*num_detection->info(), TensorInfo(TensorShape(1U), 1, DataType::F32)); in configure()
[all …]
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NEDetectionPostProcessLayer.h | 82 …tput_boxes, ITensor *output_classes, ITensor *output_scores, ITensor *num_detection, DetectionPost…
97 …*output_boxes, ITensorInfo *output_classes, ITensorInfo *output_scores, ITensorInfo *num_detection,
|
| /external/ComputeLibrary/arm_compute/runtime/CPP/functions/ |
| D | CPPDetectionPostProcessLayer.h | 69 …tput_boxes, ITensor *output_classes, ITensor *output_scores, ITensor *num_detection, DetectionPost…
84 …*output_boxes, ITensorInfo *output_classes, ITensorInfo *output_scores, ITensorInfo *num_detection,
|
| /external/ComputeLibrary/tests/validation/NEON/ |
| D | DetectionPostProcessLayer.cpp | 157 Tensor num_detection; in base_test_case() local
159 …&class_prediction, &anchors, &output_boxes, &output_classes, &output_scores, &num_detection, info); in base_test_case()
164 num_detection.allocator()->allocate(); in base_test_case()
177 validate(Accessor(num_detection), expected_num_detection, tolerance_others); in base_test_case()
|
| /external/ComputeLibrary/tests/validation/CPP/ |
| D | DetectionPostProcessLayer.cpp | 183 Tensor num_detection; in base_test_case() local
185 …&class_prediction, &anchors, &output_boxes, &output_classes, &output_scores, &num_detection, info); in base_test_case()
190 num_detection.allocator()->allocate(); in base_test_case()
203 validate(Accessor(num_detection), expected_num_detection, tolerance_others); in base_test_case()
|
| /external/ComputeLibrary/utils/ |
| D | GraphUtils.cpp | 532 const size_t num_detection = tensor.info()->valid_region().shape.y(); in access_predictions_tensor() local
535 if(num_detection > 0) in access_predictions_tensor()
543 for(size_t i = 0; i < num_detection; ++i) in access_predictions_tensor()
|