24 #ifndef ARM_COMPUTE_CL_DIRECT_CONV2D_H 25 #define ARM_COMPUTE_CL_DIRECT_CONV2D_H 76 std::unique_ptr<IClKernel> _direct_conv_kernel{
nullptr };
77 std::unique_ptr<IClKernel> _src_border_handler{
nullptr };
78 std::unique_ptr<IClKernel> _activation_kernel{
nullptr };
static Status validate(const ITensorInfo *src, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *dst, const PadStrideInfo &conv_info, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Static function to check if given info will lead to a valid configuration.
Store the tensor's metadata.
Activation Layer Information class.
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
void configure(const CLCompileContext &compile_context, ITensorInfo *src, ITensorInfo *weights, ITensorInfo *biases, ITensorInfo *dst, const PadStrideInfo &conv_info, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Set the src and dst tensors.
Padding and stride information class.
Basic function to simulate a directly convolution layer.
Basic interface for functions which have a single async CL kernel.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.