24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CL_GEMMLOWP_REDUCTION_KERNEL_H
25 #define ARM_COMPUTE_CL_GEMMLOWP_REDUCTION_KERNEL_H
void configure(const CLCompileContext &compile_context, const ITensorInfo *mtx_a, ITensorInfo *vector_sum_row, const GEMMLowpReductionKernelInfo &info) override
Initialise the kernel's input and output.
void configure(const CLCompileContext &compile_context, const ITensorInfo *mtx_b, ITensorInfo *vector_sum_col, const GEMMLowpReductionKernelInfo &info) override
Initialise the kernel's input and output.
static Status validate(const ITensorInfo *mtx_a, const ITensorInfo *vector_sum_row, const GEMMLowpReductionKernelInfo &info)
Static function to check if given info will lead to a valid configuration.
virtual void configure(const CLCompileContext &compile_context, const ITensorInfo *input, ITensorInfo *output, const GEMMLowpReductionKernelInfo &info)=0
Initialise the kernel's input and output.
static Status validate(const ITensorInfo *mtx_b, const ITensorInfo *vector_sum_col, const GEMMLowpReductionKernelInfo &info)
Static function to check if given info will lead to a valid configuration.
IClGemmLowpReductionKernel()
OpenCL kernel used to compute the row-vectors of sums of all the entries in each column of Matrix B.
void run_op(ITensorPack &tensors, const Window &window, cl::CommandQueue &queue) override
Enqueue the OpenCL kernel to process the given window on the passed OpenCL command queue.
Common interface for all the OpenCL kernels.
const Window & window() const
The maximum window the kernel can be executed on.
Describe a multidimensional execution window.
Copyright (c) 2017-2024 Arm Limited.
OpenCL kernel used to compute the row-vectors of sums of all the entries in each row of Matrix A.
Store the tensor's metadata.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(IClGemmLowpReductionKernel)
Common interface for all OpenCL reduction kernels.
void run_op(ITensorPack &tensors, const Window &window, cl::CommandQueue &queue) override
Enqueue the OpenCL kernel to process the given window on the passed OpenCL command queue.