ArmNN
 24.08
ClConvolution3dWorkload.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2021 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include "ClBaseWorkload.hpp"
9 
10 #include <arm_compute/runtime/CL/functions/CLConv3D.h>
11 #include <arm_compute/runtime/MemoryManagerOnDemand.h>
12 
13 #include <memory>
14 
15 namespace armnn
16 {
17 
19  const TensorInfo& output,
20  const Convolution3dDescriptor& descriptor,
21  const TensorInfo& weights,
22  const Optional<TensorInfo>& biases,
23  bool isFastMathEnabled = false,
24  const ActivationDescriptor* activationDescriptor = nullptr);
25 
26 class ClConvolution3dWorkload : public ClBaseWorkload<Convolution3dQueueDescriptor>
27 {
28 public:
30  const WorkloadInfo& info,
31  std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager,
32  const arm_compute::CLCompileContext& clCompileContext,
33  const bool isFastMathEnabled = false);
34  void Execute() const override;
35 
36 private:
37  mutable arm_compute::CLConv3D m_ConvolutionLayer;
38 };
39 
40 } //namespace armnn
41 
armnn::ClConvolution3dWorkload::Execute
void Execute() const override
Definition: ClConvolution3dWorkload.cpp:116
armnn::ClBaseWorkload
Definition: ClBaseWorkload.hpp:13
armnn::Convolution3dQueueDescriptor
Definition: WorkloadData.hpp:216
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition: WorkloadInfo.hpp:16
armnn::BoostLogSeverityMapping::info
@ info
armnn::Status
Status
Definition: Types.hpp:42
armnn::ClConvolution3dWorkloadValidate
arm_compute::Status ClConvolution3dWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const Convolution3dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases, bool isFastMathEnabled, const ActivationDescriptor *activationDescriptor)
Definition: ClConvolution3dWorkload.cpp:23
armnn::ClConvolution3dWorkload::ClConvolution3dWorkload
ClConvolution3dWorkload(const Convolution3dQueueDescriptor &descriptor, const WorkloadInfo &info, std::shared_ptr< arm_compute::MemoryManagerOnDemand > &memoryManager, const arm_compute::CLCompileContext &clCompileContext, const bool isFastMathEnabled=false)
Definition: ClConvolution3dWorkload.cpp:60
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
ClBaseWorkload.hpp
armnn::ClConvolution3dWorkload
Definition: ClConvolution3dWorkload.hpp:26