24 #ifndef __ARM_COMPUTE_GCSOFTMAXLAYER_H__ 25 #define __ARM_COMPUTE_GCSOFTMAXLAYER_H__ 50 GCSoftmaxLayer(std::shared_ptr<IMemoryManager> memory_manager =
nullptr);
Base class for all functions.
Interface for GLES Compute tensor.
Basic function to compute a SoftmaxLayer.
Interface for OpenGL ES tensor.
This file contains all available output stages for GEMMLowp on OpenCL.
Interface for the identifying the max value of 1D Logits.
void configure(const IGCTensor *input, IGCTensor *output, float beta=1.0f)
Set the input and output tensors.
GCSoftmaxLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Constructor.
Interface for calculating the final step of the Softmax Layer where each logit value is multiplied by...
void run() override
Run the kernels contained in the function.
Interface for shifting the logits values around the max value and exponentiating the result...