| 369 | } |
| 370 | |
| 371 | void NEGenerateProposalsLayer::run() |
| 372 | { |
| 373 | ARM_COMPUTE_TRACE_EVENT(ARM_COMPUTE_PROF_CAT_CPU, ARM_COMPUTE_PROF_LVL_CPU, "NEGenerateProposalsLayer::run"); |
| 374 | // Acquire all the temporaries |
| 375 | MemoryGroupResourceScope scope_mg(_memory_group); |
| 376 | |
| 377 | // Compute all the anchors |
| 378 | NEScheduler::get().schedule(_compute_anchors.get(), Window::DimY); |
| 379 | |
| 380 | // Transpose and reshape the inputs |
| 381 | if (!_is_nhwc) |
| 382 | { |
| 383 | _permute_deltas.run(); |
| 384 | _permute_scores.run(); |
| 385 | } |
| 386 | |
| 387 | _flatten_deltas.run(); |
| 388 | _flatten_scores.run(); |
| 389 | |
| 390 | if (_is_qasymm8) |
| 391 | { |
| 392 | _dequantize_anchors.run(); |
| 393 | _dequantize_deltas.run(); |
| 394 | } |
| 395 | |
| 396 | // Build the boxes |
| 397 | _bounding_box.run(); |
| 398 | |
| 399 | if (_is_qasymm8) |
| 400 | { |
| 401 | _quantize_all_proposals.run(); |
| 402 | } |
| 403 | |
| 404 | // Non maxima suppression |
| 405 | _cpp_nms.run(); |
| 406 | |
| 407 | // Add dummy batch indexes |
| 408 | _pad.run(); |
| 409 | } |
| 410 | } // namespace arm_compute |