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