24 #ifndef __ARM_COMPUTE_CLSOFTMAXLAYER_H__ 25 #define __ARM_COMPUTE_CLSOFTMAXLAYER_H__ 53 CLSoftmaxLayer(std::shared_ptr<IMemoryManager> memory_manager =
nullptr);
Interface for max, shifting, exponentiating and summing the logits.
Base class for all functions.
Basic function to compute a SoftmaxLayer.
Store the tensor's metadata.
This file contains all available output stages for GEMMLowp on OpenCL.
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.
static Status validate(const ITensorInfo *input, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of CLSoftmaxLayer.
void configure(const ICLTensor *input, ICLTensor *output, float beta=1.0f)
Set the input and output tensors.
Interface for OpenCL tensor.
CLSoftmaxLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Constructor.
Basic implementation of the OpenCL tensor interface.