38 ITensorPack select_activation_src_dst(ITensorPack &tensors)
51 auto k = std::make_unique<kernels::ClDirectConvolutionKernel>();
54 _direct_conv_kernel = std::move(k);
62 auto b = std::make_unique<CLFillBorderKernel>();
64 _src_border_handler = std::move(
b);
68 auto a = std::make_unique<kernels::ClActivationKernel>();
69 a->configure(compile_context,
dst,
dst, act_info);
70 _activation_kernel = std::move(a);
95 if(_activation_kernel)
97 auto act_pack = select_activation_src_dst(tensors);
Class describing the value of a pixel for any image format.
bool enabled() const
Check if initialised.
static CLScheduler & get()
Access the scheduler singleton.
#define ARM_COMPUTE_RETURN_ON_ERROR(status)
Checks if a status contains an error and returns it.
static Status validate(const ITensorInfo *src, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *dst, const PadStrideInfo &conv_info, const GPUTarget target)
Static function to check if given info will lead to a valid configuration of ClDirectConvolutionKerne...
Store the tensor's metadata.
Activation Layer Information class.
SimpleTensor< float > src
Copyright (c) 2017-2021 Arm Limited.
Interface to enqueue OpenCL kernels and get/set the OpenCL CommandQueue and ICLTuner.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
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 of ClDirectConvolution.
void enqueue_op(ICLKernel &kernel, ITensorPack &tensors, bool flush=true)
Schedule the execution of the passed kernel if possible.
static Status validate(const ITensorInfo *src, const ITensorInfo *dst, const ActivationLayerInfo &act_info)
Static function to check if given info will lead to a valid configuration of ClActivationKernel.
Padding and stride information class.
bool is_data_type_quantized_asymmetric(DataType dt)
Check if a given data type is of asymmetric quantized type.
void tune_kernel_static(ICLKernel &kernel)
Tunes OpenCL kernel.
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.