| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLGenerateProposalsLayer.cpp | 74 …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()
186 auto_init_if_empty(*num_valid_proposals->info(), TensorShape(1), 1, DataType::U32); in configure()
194 _num_valid_proposals = num_valid_proposals; in configure()
197 …, &_proposals_4_roi_values, &_classes_nms_unused, nullptr, &_keeps_nms_unused, num_valid_proposals, in configure()
210 … const ITensorInfo *num_valid_proposals, const GenerateProposalsInfo &info) in validate() argument
212 …MPUTE_RETURN_ERROR_ON_NULLPTR(scores, deltas, anchors, proposals, scores_out, num_valid_proposals); in validate()
[all …]
|
| /external/ComputeLibrary/src/runtime/NEON/functions/ |
| D | NEGenerateProposalsLayer.cpp | 70 …tas, const ITensor *anchors, ITensor *proposals, ITensor *scores_out, ITensor *num_valid_proposals, in configure() argument
73 … ARM_COMPUTE_ERROR_ON_NULLPTR(scores, deltas, anchors, proposals, scores_out, num_valid_proposals); in configure()
74 …s->info(), anchors->info(), proposals->info(), scores_out->info(), num_valid_proposals->info(), in… in configure()
175 auto_init_if_empty(*num_valid_proposals->info(), TensorShape(1), 1, DataType::U32); in configure()
183 _num_valid_proposals = num_valid_proposals; in configure()
196 num_valid_proposals /* keeps_size*/, in configure()
210 … const ITensorInfo *num_valid_proposals, const GenerateProposalsInfo &info) in validate() argument
212 …MPUTE_RETURN_ERROR_ON_NULLPTR(scores, deltas, anchors, proposals, scores_out, num_valid_proposals); in validate()
286 if(num_valid_proposals->total_size() > 0) in validate()
288 ARM_COMPUTE_RETURN_ERROR_ON(num_valid_proposals->num_dimensions() > 1); in validate()
[all …]
|
| /external/ComputeLibrary/tests/validation/CL/ |
| D | GenerateProposalsLayer.cpp | 138 …scores, deltas, anchors, proposals, scores_out, num_valid_proposals, generate_proposals_info, expe…
145 … &num_valid_proposals.clone()->set_is_resizable(true),
333 CLTensor num_valid_proposals; variable
335 num_valid_proposals.allocator()->init(TensorInfo(TensorShape(1), 1, DataType::U32));
338 …proposals.configure(&scores, &bbox_deltas, &anchors, &proposals, &scores_out, &num_valid_proposals,
346 num_valid_proposals.allocator()->allocate();
356 num_valid_proposals.map();
357 …const uint32_t N = *reinterpret_cast<uint32_t *>(num_valid_proposals.ptr_to_element(Coordinates(0,…
358 num_valid_proposals.unmap();
|
| /external/ComputeLibrary/tests/validation/NEON/ |
| D | GenerateProposalsLayer.cpp | 138 …scores, deltas, anchors, proposals, scores_out, num_valid_proposals, generate_proposals_info, expe…
145 … &num_valid_proposals.clone()->set_is_resizable(true),
332 Tensor num_valid_proposals; variable
334 num_valid_proposals.allocator()->init(TensorInfo(TensorShape(1), 1, DataType::U32));
337 …proposals.configure(&scores, &bbox_deltas, &anchors, &proposals, &scores_out, &num_valid_proposals,
345 num_valid_proposals.allocator()->allocate();
355 …const uint32_t N = *reinterpret_cast<uint32_t *>(num_valid_proposals.ptr_to_element(Coordinates(0,…
|
| /external/ComputeLibrary/arm_compute/runtime/CL/functions/ |
| D | CLGenerateProposalsLayer.h | 93 …st ICLTensor *anchors, ICLTensor *proposals, ICLTensor *scores_out, ICLTensor *num_valid_proposals,
112 ICLTensor *num_valid_proposals, const GenerateProposalsInfo &info);
129 const ITensorInfo *num_valid_proposals,
|
| /external/ComputeLibrary/arm_compute/runtime/NEON/functions/ |
| D | NEGenerateProposalsLayer.h | 88 …tas, const ITensor *anchors, ITensor *proposals, ITensor *scores_out, ITensor *num_valid_proposals,
106 const ITensorInfo *num_valid_proposals,
|
| /external/ComputeLibrary/arm_compute/graph/backends/ |
| D | ValidateHelpers.h | 340 arm_compute::ITensorInfo *num_valid_proposals = get_backing_tensor_info(node.output(2)); in validate_generate_proposals_layer() local
343 …roposalsLayer::validate(scores, deltas, anchors, proposals, scores_out, num_valid_proposals, info); in validate_generate_proposals_layer()
|
| D | FunctionHelpers.h | 1064 …typename TargetInfo::TensorType *num_valid_proposals = get_backing_tensor<TargetInfo>(node.output(… in create_generate_proposals_layer() local
1075 func->configure(scores, deltas, anchors, proposals, scores_out, num_valid_proposals, info); in create_generate_proposals_layer()
1085 … << " Num valid proposals shape: " << num_valid_proposals->info()->tensor_shape() in create_generate_proposals_layer()
|