24 #ifndef ARM_COMPUTE_CLARGMINMAXLAYERKERNEL_H 25 #define ARM_COMPUTE_CLARGMINMAXLAYERKERNEL_H 102 unsigned int _reduction_axis;
~CLArgMinMaxLayerKernel()=default
Default destructor.
const Window & window() const
The maximum window the kernel can be executed on.
ReductionOperation
Available reduction operations.
CLArgMinMaxLayerKernel & operator=(const CLArgMinMaxLayerKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
void configure(const ICLTensor *input, const ICLTensor *prev_output, ICLTensor *output, unsigned int axis, ReductionOperation op)
Set the input and output tensors.
Interface for the reduction operation kernel.
Store the tensor's metadata.
Common interface for all the OpenCL kernels.
Copyright (c) 2017-2021 Arm Limited.
CLArgMinMaxLayerKernel()
Default constructor.
static Status validate(const ITensorInfo *input, const ITensorInfo *prev_output, const ITensorInfo *output, unsigned int axis, ReductionOperation op)
Static function to check if given info will lead to a valid configuration of CLArgMinMaxLayerKernel.
Interface for OpenCL tensor.
void run(const Window &window, cl::CommandQueue &queue) override
Enqueue the OpenCL kernel to process the given window on the passed OpenCL command queue...
Describe a multidimensional execution window.