24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CLARGMINMAXLAYERKERNEL_H
25 #define ARM_COMPUTE_CLARGMINMAXLAYERKERNEL_H
101 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.
CLArgMinMaxLayerKernel()
Default constructor.
Interface for OpenCL tensor.
ReductionOperation
Available reduction operations.
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 CLArgMinMaxLayerKernel.
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.
Interface for the reduction operation kernel.
void configure(const ICLTensor *input, ICLTensor *output, unsigned int axis, ReductionOperation op)
Set the input and output tensors.
~CLArgMinMaxLayerKernel()=default
Default destructor.
Store the tensor's metadata.
CLArgMinMaxLayerKernel & operator=(const CLArgMinMaxLayerKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers)