24 #ifndef ARM_COMPUTE_GCSOFTMAXLAYERKERNEL_H 25 #define ARM_COMPUTE_GCSOFTMAXLAYERKERNEL_H const Window & window() const
The maximum window the kernel can be executed on.
DATA_TYPE sum(__global const DATA_TYPE *input)
Calculate sum of a vector.
Interface for GLES Compute tensor.
Copyright (c) 2017-2021 Arm Limited.
Interface for simple GLES kernels having 1 tensor input and 1 tensor output.
Interface for the identifying the max value of 1D Logits.
Interface for calculating the final step of the Softmax Layer where each logit value is multiplied by...
Common interface for all the GLES kernels.
Interface for shifting the logits values around the max value and exponentiating the result...
void configure(const IGCTensor *input, IGCTensor *output)
Set the input and output tensors.
Describe a multidimensional execution window.
IGCSimpleKernel & operator=(const IGCSimpleKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
void run(const Window &window) override
Enqueue the OpenGL ES shader to process the given window.