24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CLREDUCTIONOPERATIONKERNEL_H
25 #define ARM_COMPUTE_CLREDUCTIONOPERATIONKERNEL_H
96 unsigned int _reduction_axis;
void run(const Window &window, cl::CommandQueue &queue) override
Enqueue the OpenCL kernel to process the given window on the passed OpenCL command queue.
Interface for OpenCL tensor.
Interface for the reduction operation kernel.
ReductionOperation
Available reduction operations.
void configure(const ICLTensor *input, ICLTensor *output, unsigned int axis, ReductionOperation op)
Set the input and output tensors.
CLReductionOperationKernel()
Default constructor.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, unsigned int axis, ReductionOperation op)
Static function to check if given info will lead to a valid configuration of CLReductionOperationKern...
~CLReductionOperationKernel()=default
Default destructor.
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.
CLReductionOperationKernel & operator=(const CLReductionOperationKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
Store the tensor's metadata.