Interface for performing a L2 normalize on a given axis given the square sum of it in this axis.
More...
#include <NEL2NormalizeLayerKernel.h>
Interface for performing a L2 normalize on a given axis given the square sum of it in this axis.
Definition at line 34 of file NEL2NormalizeLayerKernel.h.
◆ NEL2NormalizeLayerKernel() [1/3]
◆ NEL2NormalizeLayerKernel() [2/3]
Prevent instances of this class from being copied (As this class contains pointers)
◆ NEL2NormalizeLayerKernel() [3/3]
Allow instances of this class to be moved.
◆ ~NEL2NormalizeLayerKernel()
◆ configure()
Set the input and output tensors.
- Parameters
-
[in] | input | Source tensor. Data types supported: F16/F32. |
[in] | sum | Sum values tensor. Data types supported: same as input . Sum will have the same number of dimensions as input. |
[out] | output | Destination tensor. Data types and data layouts supported: same as input . Output will have the same number of dimensions as input. |
[in] | axis | Axis along which to reduce. Negative values wrap around. Maximum supported actual reduction axis : 2 |
[in] | epsilon | Lower bound value for the normalization. |
◆ name()
const char* name |
( |
| ) |
const |
|
inlineoverridevirtual |
◆ operator=() [1/2]
Prevent instances of this class from being copied (As this class contains pointers)
◆ operator=() [2/2]
Allow instances of this class to be moved.
◆ run()
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()) |
[in] | info | Info about executing thread and CPU. |
Reimplemented from ICPPKernel.
◆ validate()
Static function to check if given info will lead to a valid configuration of NEL2NormalizeLayerKernel.
- Parameters
-
[in] | input | Source tensor info. Data types supported: F16/F32. |
[in] | sum | Sum values tensor info. Data types supported: same as input . Sum will have the same number of dimensions as input. |
[in] | output | Destination tensor info. Data types and data layouts supported: same as input . Output will have the same number of dimensions as input. |
[in] | axis | Axis along which to reduce. Negative values wrap around. Maximum supported actual reduction axis : 2 |
[in] | epsilon | Lower bound value for the normalization. |
- Returns
- a status
Referenced by NEL2NormalizeLayer::validate().
The documentation for this class was generated from the following file: