diff options
Diffstat (limited to 'src/backends/gpuFsa/workloads/GpuFsaConstantWorkload.hpp')
-rw-r--r-- | src/backends/gpuFsa/workloads/GpuFsaConstantWorkload.hpp | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/src/backends/gpuFsa/workloads/GpuFsaConstantWorkload.hpp b/src/backends/gpuFsa/workloads/GpuFsaConstantWorkload.hpp new file mode 100644 index 0000000000..98b383b89f --- /dev/null +++ b/src/backends/gpuFsa/workloads/GpuFsaConstantWorkload.hpp @@ -0,0 +1,30 @@ +// +// Copyright © 2024 Arm Ltd and Contributors. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#pragma once + +#include "GpuFsaBaseWorkload.hpp" + +#include <arm_compute/core/Error.h> +#include <arm_compute/core/CL/CLCompileContext.h> + +namespace armnn +{ + arm_compute::Status GpuFsaConstantWorkloadValidate(const TensorInfo& output); + + class GpuFsaConstantWorkload : public GpuFsaBaseWorkload<ConstantQueueDescriptor> + { + public: + GpuFsaConstantWorkload(const ConstantQueueDescriptor& descriptor, + const WorkloadInfo& info, + const arm_compute::CLCompileContext& clCompileContext); + + void Execute() const override; + + private: + mutable bool m_RanOnce; + }; + +} //namespace armnn |