Home
last modified time | relevance | path

Searched refs:num_valid_proposals (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLGenerateProposalsLayer.cpp74 …st ICLTensor *anchors, ICLTensor *proposals, ICLTensor *scores_out, ICLTensor *num_valid_proposals, in configure() argument
77 ….get_compile_context(), scores, deltas, anchors, proposals, scores_out, num_valid_proposals, info); in configure()
82 … ICLTensor *num_valid_proposals, const GenerateProposalsInfo &info) in configure() argument
84 … ARM_COMPUTE_ERROR_ON_NULLPTR(scores, deltas, anchors, proposals, scores_out, num_valid_proposals); in configure()
85 …s->info(), anchors->info(), proposals->info(), scores_out->info(), num_valid_proposals->info(), in… in configure()
86 … ARM_COMPUTE_LOG_PARAMS(scores, deltas, anchors, proposals, scores_out, num_valid_proposals, info); in configure()
187 auto_init_if_empty(*num_valid_proposals->info(), TensorShape(1), 1, DataType::U32); in configure()
195 _num_valid_proposals = num_valid_proposals; in configure()
198 …, &_proposals_4_roi_values, &_classes_nms_unused, nullptr, &_keeps_nms_unused, num_valid_proposals, in configure()
211 … const ITensorInfo *num_valid_proposals, const GenerateProposalsInfo &info) in validate() argument
[all …]
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNEGenerateProposalsLayer.cpp71 …tas, const ITensor *anchors, ITensor *proposals, ITensor *scores_out, ITensor *num_valid_proposals, in configure() argument
74 … ARM_COMPUTE_ERROR_ON_NULLPTR(scores, deltas, anchors, proposals, scores_out, num_valid_proposals); in configure()
75 …s->info(), anchors->info(), proposals->info(), scores_out->info(), num_valid_proposals->info(), in… in configure()
76 … ARM_COMPUTE_LOG_PARAMS(scores, deltas, anchors, proposals, scores_out, num_valid_proposals, info); in configure()
178 auto_init_if_empty(*num_valid_proposals->info(), TensorShape(1), 1, DataType::U32); in configure()
186 _num_valid_proposals = num_valid_proposals; in configure()
199 num_valid_proposals /* keeps_size*/, in configure()
213 … const ITensorInfo *num_valid_proposals, const GenerateProposalsInfo &info) in validate() argument
215 …MPUTE_RETURN_ERROR_ON_NULLPTR(scores, deltas, anchors, proposals, scores_out, num_valid_proposals); in validate()
289 if(num_valid_proposals->total_size() > 0) in validate()
[all …]
/aosp_15_r20/external/ComputeLibrary/tests/validation/CL/
H A DGenerateProposalsLayer.cpp141 …scores, deltas, anchors, proposals, scores_out, num_valid_proposals, generate_proposals_info, expe…
148 … &num_valid_proposals.clone()->set_is_resizable(true),
336 CLTensor num_valid_proposals; variable
338 num_valid_proposals.allocator()->init(TensorInfo(TensorShape(1), 1, DataType::U32));
341 …proposals.configure(&scores, &bbox_deltas, &anchors, &proposals, &scores_out, &num_valid_proposals,
349 num_valid_proposals.allocator()->allocate();
359 num_valid_proposals.map();
360 …const uint32_t N = *reinterpret_cast<uint32_t *>(num_valid_proposals.ptr_to_element(Coordinates(0,…
361 num_valid_proposals.unmap();
/aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/
H A DGenerateProposalsLayer.cpp141 …scores, deltas, anchors, proposals, scores_out, num_valid_proposals, generate_proposals_info, expe…
148 … &num_valid_proposals.clone()->set_is_resizable(true),
335 Tensor num_valid_proposals; variable
337 num_valid_proposals.allocator()->init(TensorInfo(TensorShape(1), 1, DataType::U32));
340 …proposals.configure(&scores, &bbox_deltas, &anchors, &proposals, &scores_out, &num_valid_proposals,
348 num_valid_proposals.allocator()->allocate();
358 …const uint32_t N = *reinterpret_cast<uint32_t *>(num_valid_proposals.ptr_to_element(Coordinates(0,…
/aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/CL/functions/
H A DCLGenerateProposalsLayer.h103 …st ICLTensor *anchors, ICLTensor *proposals, ICLTensor *scores_out, ICLTensor *num_valid_proposals,
122 ICLTensor *num_valid_proposals, const GenerateProposalsInfo &info);
139 const ITensorInfo *num_valid_proposals,
/aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/NEON/functions/
H A DNEGenerateProposalsLayer.h98 …tas, const ITensor *anchors, ITensor *proposals, ITensor *scores_out, ITensor *num_valid_proposals,
116 const ITensorInfo *num_valid_proposals,
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/backends/
H A DValidateHelpers.h376 arm_compute::ITensorInfo *num_valid_proposals = get_backing_tensor_info(node.output(2)); in validate_generate_proposals_layer() local
379 …roposalsLayer::validate(scores, deltas, anchors, proposals, scores_out, num_valid_proposals, info); in validate_generate_proposals_layer()
H A DFunctionHelpers.h1244 …typename TargetInfo::TensorType *num_valid_proposals = get_backing_tensor<TargetInfo>(node.output(… in create_generate_proposals_layer() local
1255 func->configure(scores, deltas, anchors, proposals, scores_out, num_valid_proposals, info); in create_generate_proposals_layer()
1265 … << " Num valid proposals shape: " << num_valid_proposals->info()->tensor_shape() in create_generate_proposals_layer()