ArmNN
 25.11
Loading...
Searching...
No Matches
ClFillWorkload.hpp
Go to the documentation of this file.
1//
2// Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
3// SPDX-License-Identifier: MIT
4//
5
6#pragma once
7
9#include "ClBaseWorkload.hpp"
10#include <arm_compute/runtime/CL/functions/CLFill.h>
11
12namespace armnn {
13
14class ClFillWorkload : public ClBaseWorkload<FillQueueDescriptor>
15{
16public:
17 ClFillWorkload(const FillQueueDescriptor& descriptor,
18 const WorkloadInfo& info,
19 const arm_compute::CLCompileContext& clCompileContext);
20 void Execute() const override;
21
22private:
23 mutable arm_compute::CLFill m_Layer;
24};
25
26} //namespace armnn
ClBaseWorkload(const FillQueueDescriptor &descriptor, const WorkloadInfo &info)
ClFillWorkload(const FillQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
void Execute() const override
Copyright (c) 2021 ARM Limited and Contributors.
Contains information about TensorInfos of a layer.