Interface for shifting the logits values around the max value and exponentiating the result.
More...
#include <NESoftmaxLayerKernel.h>
Interface for shifting the logits values around the max value and exponentiating the result.
Definition at line 56 of file NESoftmaxLayerKernel.h.
Prevent instances of this class from being copied (As this class contains pointers)
Allow instances of this class to be moved.
The size of the border for that kernel.
- Returns
- The width in number of elements of the border.
Reimplemented from IKernel.
Set the input and output tensors.
- Parameters
-
[in] | input | Source tensor. Data types supported: F32. |
[in] | max | Max values tensor. Data types supported: same as input . |
[out] | output | Destination tensor. Data types supported: same as input . |
[out] | sum | Sum of 1D logits tensor. Data types supported: same as input . |
Prevent instances of this class from being copied (As this class contains pointers)
Allow instances of this class to be moved.
void run |
( |
const Window & |
window | ) |
|
|
overridevirtual |
Execute the kernel on the passed window.
- Warning
- If is_parallelisable() returns false then the passed window must be equal to window()
- Note
- The window has to be a region within the window returned by the window() method
-
The width of the window has to be a multiple of num_elems_processed_per_iteration().
- Parameters
-
[in] | window | Region on which to execute the kernel. (Must be a region of the window returned by window()) |
Implements ICPPKernel.
The documentation for this class was generated from the following file: