Interface for the normalization layer kernel.
More...
#include <NENormalizationLayerKernel.h>
Interface for the normalization layer kernel.
Definition at line 35 of file NENormalizationLayerKernel.h.
Prevent instances of this class from being copied (As this class contains pointers)
Default Move Constructor.
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. 3 lower dims represent a single input with dimensions [width, height, IFM], and an optional 4th dimension for batch of inputs. Data types supported: F32. |
[in] | input_squared | Source with each element has been squared. 3 lower dims represent a single input with dimensions [width, height, IFM], Data type supported: same as input |
[out] | output | Destination tensor. Output will have the same number of dimensions as input. Data type supported: same as input |
[in] | norm_info | Normalization layer information like the normalization type, normalization size and other parameters. |
Prevent instances of this class from being copied (As this class contains pointers)
Default move assignment operator.
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: