13 #include <arm_compute/runtime/CL/functions/CLConvolutionLayer.h> 14 #include <arm_compute/runtime/MemoryManagerOnDemand.h> 22 const TensorInfo& output,
23 const Convolution2dDescriptor& descriptor,
24 const TensorInfo& weights,
25 const Optional<TensorInfo>& biases,
26 bool isFastMathEnabled =
false,
27 const ActivationDescriptor* activationDescriptor =
nullptr);
34 std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager,
35 const arm_compute::CLCompileContext& clCompileContext,
36 const bool isFastMathEnabled =
false);
42 mutable arm_compute::CLConvolutionLayer m_ConvolutionLayer;
44 std::unique_ptr<arm_compute::CLTensor> m_KernelTensor;
45 std::unique_ptr<arm_compute::CLTensor> m_BiasTensor;
47 arm_compute::ConvolutionMethod m_ConvolutionMethod;
49 void FreeUnusedTensors();
Copyright (c) 2021 ARM Limited and Contributors.
arm_compute::Status ClConvolution2dWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const Convolution2dDescriptor &descriptor, const TensorInfo &weights, const Optional< TensorInfo > &biases, bool isFastMathEnabled, const ActivationDescriptor *activationDescriptor)
arm_compute::ConvolutionMethod GetConvolutionMethod() const
ClConvolution2dWorkload(const Convolution2dQueueDescriptor &descriptor, const WorkloadInfo &info, std::shared_ptr< arm_compute::MemoryManagerOnDemand > &memoryManager, const arm_compute::CLCompileContext &clCompileContext, const bool isFastMathEnabled=false)
void Execute() const override
Contains information about TensorInfos of a layer.