24 #ifndef ARM_COMPUTE_CL_GEMMCONVOLUTION_H 25 #define ARM_COMPUTE_CL_GEMMCONVOLUTION_H 40 class ClGemmLowpMatrixMultiplyCore;
45 class ClWeightsReshapeKernel;
46 class ClActivationKernel;
163 std::unique_ptr<kernels::ClWeightsReshapeKernel> _weights_reshape_kernel;
164 std::unique_ptr<kernels::ClIm2ColKernel> _im2col_kernel;
165 std::unique_ptr<ClGemm> _mm_gemm;
166 std::unique_ptr<ClGemmLowpMatrixMultiplyCore> _mm_gemmlowp;
167 std::unique_ptr<opencl::kernels::ClCol2ImKernel> _col2im_kernel;
168 std::unique_ptr<kernels::ClActivationKernel> _activation_kernel;
177 bool _fuse_activation;
Store the tensor's metadata.
Basic function to compute the convolution layer.
Activation Layer Information class.
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
std::vector< MemoryInfo > MemoryRequirements
Convolution Layer Weights Information class.
GEMMLowp output stage info.
Descriptor used by the Convolution function.
Store the tensor's metadata.
im2col_func configure(src_target.info(), dst_target.info(), spatial_kernel, conv_info, has_bias)
Basic interface for functions which have a single async CL kernel.
Status validate(const ITensorInfo *scores_in, const ITensorInfo *boxes_in, const ITensorInfo *batch_splits_in, const ITensorInfo *scores_out, const ITensorInfo *boxes_out, const ITensorInfo *classes, const ITensorInfo *batch_splits_out, const ITensorInfo *keeps, const ITensorInfo *keeps_size, const BoxNMSLimitInfo info)