24 #ifndef __ARM_COMPUTE_NESOFTMAXLAYERKERNEL_H__ 25 #define __ARM_COMPUTE_NESOFTMAXLAYERKERNEL_H__
+
Container for 2D border size.
Common interface for all kernels implemented in C++.
-
Interface for shifting the logits values around the max value and exponentiating the result...
+
Interface for shifting the logits values around the max value and exponentiating the result...
+
NELogits1DMaxKernel()
Default constructor.
Interface for simple NEON kernels having 1 tensor input and 1 tensor output.
Interface for NEON tensor.
Interface for the identifying the max value of 1D Logits.
void configure(const ITensor *input, ITensor *output)
Set the input and output tensors.
void run(const Window &window) override
Execute the kernel on the passed window.
+
BorderSize border_size() const override
The size of the border for that kernel.
ICPPSimpleKernel & operator=(const ICPPSimpleKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
const Window & window() const
The maximum window the kernel can be executed on.
Describe a multidimensional execution window.
-
Interface for calculating the final step of the Softmax Layer where each logit value is multiplied by...
-
NELogits1DMaxKernel()=default
Default constructor.
+
Interface for calculating the final step of the Softmax Layer where each logit value is multiplied by...