patch/latest/_ref_detection_post_process_workload_8hpp_source.xhtml

 //

- // Copyright © 2022 Arm Ltd and Contributors. All rights reserved.

- // SPDX-License-Identifier: MIT

- //

-

- #pragma once

-

- #include "RefBaseWorkload.hpp"

- #include <armnn/backends/WorkloadData.hpp>

-

- namespace armnn

- {

-

- class RefDetectionPostProcessWorkload : public RefBaseWorkload<DetectionPostProcessQueueDescriptor>

- {

- public:

-     explicit RefDetectionPostProcessWorkload(const DetectionPostProcessQueueDescriptor& descriptor,

-                                              const WorkloadInfo& info);

-     void Execute() const override;

-     void ExecuteAsync(ExecutionData& executionData)  override;

-

- private:

-     void Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const;

-     std::unique_ptr<ScopedTensorHandle> m_Anchors;

- };

-

- } //namespace armnn

-