24 #ifndef __ARM_COMPUTE_CLSOFTMAXLAYER_H__ 25 #define __ARM_COMPUTE_CLSOFTMAXLAYER_H__ Base class for all functions.
Basic function to compute a SoftmaxLayer.
Interface for calculating the final step of the Softmax Layer where each logit value is multiplied by...
Interface for shifting the logits values around the max value and exponentiating the result...
void run() override
Run the kernels contained in the function.
void configure(const ICLTensor *input, ICLTensor *output)
Set the input and output tensors.
CLSoftmaxLayer()
Constructor.
Interface for the identifying the max value of 1D Logits.
Interface for OpenCL tensor.
Basic implementation of the OpenCL tensor interface.