24 #ifndef __ARM_COMPUTE_CLSOFTMAXLAYERKERNEL_H__ 25 #define __ARM_COMPUTE_CLSOFTMAXLAYERKERNEL_H__
ICLSimpleKernel & operator=(const ICLSimpleKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers).
Common interface for all the OpenCL kernels.
Interface for calculating the final step of the Softmax Layer where each logit value is multiplied by...
Interface for simple OpenCL kernels having 1 tensor input and 1 tensor output.
Interface for shifting the logits values around the max value and exponentiating the result...
Interface for the identifying the max value of 1D Logits.
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...
void configure(const ICLTensor *input, ICLTensor *output)
Set the input and output tensors.
const Window & window() const
The maximum window the kernel can be executed on.
Describe a multidimensional execution window.