24.02.1
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_GPU_CL_OPERATORS_CLGEMMCONV2D_H
25 #define ACL_SRC_GPU_CL_OPERATORS_CLGEMMCONV2D_H
41 class ClGemmLowpMatrixMultiplyCore;
46 class ClWeightsReshapeKernel;
47 class ClActivationKernel;
184 std::unique_ptr<kernels::ClWeightsReshapeKernel> _weights_reshape_kernel;
185 std::unique_ptr<kernels::ClIm2ColKernel> _im2col_kernel;
186 std::unique_ptr<ClGemm> _mm_gemm;
187 std::unique_ptr<ClGemmLowpMatrixMultiplyCore> _mm_gemmlowp;
188 std::unique_ptr<opencl::kernels::ClCol2ImKernel> _col2im_kernel;
189 std::unique_ptr<kernels::ClActivationKernel> _activation_kernel;
198 bool _fuse_activation;
206 #endif // ACL_SRC_GPU_CL_OPERATORS_CLGEMMCONV2D_H
~ClGemmConv2d()
Default destructor.
std::vector< MemoryInfo > MemoryRequirements
ClGemmConv2d & operator=(const ClGemmConv2d &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
Convolution Layer Weights Information class.
SimpleTensor< float > src
GEMMLowp output stage info.
Basic function to compute the convolution layer.
ClGemmConv2d()
Constructor.
Descriptor used by the 2d Convolution function.
Activation Layer Information class.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
Basic interface for functions which have a single async CL kernel.
experimental::MemoryRequirements workspace() const override
Return the memory requirements required by the workspace.
void configure(const ClCompileContext &compile_context, ITensorInfo *src, ITensorInfo *weights, ITensorInfo *biases, ITensorInfo *dst, const Conv2dInfo &conv2d_info, const WeightsInfo &weights_info=WeightsInfo())
Set the input and output tensors.
static Status validate(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *output, const Conv2dInfo &conv2d_info, const WeightsInfo &weights_info=WeightsInfo())
Static function to check if given info will lead to a valid configuration.
Store the tensor's metadata.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
void prepare(ITensorPack &constants) override
Prepare the function for executing.