Searched refs:input_class_score (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CPP/functions/ |
| H A D | CPPDetectionPostProcessLayer.cpp | 41 …arguments(const ITensorInfo *input_box_encoding, const ITensorInfo *input_class_score, const ITens… in validate_arguments() argument
45 ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR(input_box_encoding, input_class_score, input_anchors); in validate_arguments()
54 ARM_COMPUTE_RETURN_ERROR_ON_MSG(input_class_score->dimension(0) != (info.num_classes() + 1), in validate_arguments()
62 …ARM_COMPUTE_RETURN_ERROR_ON_MSG((input_box_encoding->dimension(1) != input_class_score->dimension(… in validate_arguments()
272 …:validate(const ITensorInfo *input_box_encoding, const ITensorInfo *input_class_score, const ITens… in validate() argument
283 …ARM_COMPUTE_RETURN_ON_ERROR(validate_arguments(input_box_encoding, input_class_score, input_anchor… in validate()
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| H A D | NEDetectionPostProcessLayer.h | 96 … validate(const ITensorInfo *input_box_encoding, const ITensorInfo *input_class_score, const ITens…
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/CPP/functions/ |
| H A D | CPPDetectionPostProcessLayer.h | 83 … validate(const ITensorInfo *input_box_encoding, const ITensorInfo *input_class_score, const ITens…
|