24 #ifndef ARM_COMPUTE_GCSOFTMAXLAYER_H 25 #define ARM_COMPUTE_GCSOFTMAXLAYER_H 53 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.
Copyright (c) 2017-2021 Arm Limited.
Interface for the identifying the max value of 1D Logits.
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...
void configure(const IGCTensor *input, IGCTensor *output, float beta=1.0f, int32_t axis=0)
Set the input and output tensors.