#include <CLInstanceNormalizationLayerKernelEx.h>
|
| static Status | validate (const ITensorInfo *input, const ITensorInfo *output, const ITensorInfo *gamma=nullptr, const ITensorInfo *beta=nullptr, float epsilon=1e-12f) |
| |
Interface for performing an instance normalization
Definition at line 51 of file CLInstanceNormalizationLayerKernelEx.h.
◆ CLInstanceNormalizationLayerKernelEx() [1/3]
| arm_compute::CLInstanceNormalizationLayerKernelEx::CLInstanceNormalizationLayerKernelEx |
( |
| ) |
|
◆ CLInstanceNormalizationLayerKernelEx() [2/3]
Prevent instances of this class from being copied (As this class contains pointers)
◆ CLInstanceNormalizationLayerKernelEx() [3/3]
Default Move Constructor.
◆ ~CLInstanceNormalizationLayerKernelEx()
| arm_compute::CLInstanceNormalizationLayerKernelEx::~CLInstanceNormalizationLayerKernelEx |
( |
| ) |
|
|
default |
◆ configure()
| void arm_compute::CLInstanceNormalizationLayerKernelEx::configure |
( |
ICLTensor * |
input, |
|
|
ICLTensor * |
output, |
|
|
ICLTensor * |
gamma = nullptr, |
|
|
ICLTensor * |
beta = nullptr, |
|
|
float |
epsilon = 1e-12f |
|
) |
| |
Set the input and output tensors.
- Parameters
-
| [in,out] | input | Source tensor. Data types supported: F16/F32. Data layout supported: NCHW |
| [out] | output | Destination tensor. Data types and data layouts supported: same as input. |
| [in] | gamma | (Optional) The scale tensor applied to the normalized tensor. Defaults to nullptr |
| [in] | beta | (Optional) The offset tensor applied to the normalized tensor. Defaults to nullptr |
| [in] | epsilon | (Optional) Lower bound value for the normalization. Defaults to 1e-12 |
◆ operator=() [1/2]
Default move assignment operator
◆ operator=() [2/2]
Prevent instances of this class from being copied (As this class contains pointers)
◆ run()
| void arm_compute::CLInstanceNormalizationLayerKernelEx::run |
( |
const Window & |
window, |
|
|
cl::CommandQueue & |
queue |
|
) |
| |
|
override |
◆ validate()
| static Status arm_compute::CLInstanceNormalizationLayerKernelEx::validate |
( |
const ITensorInfo * |
input, |
|
|
const ITensorInfo * |
output, |
|
|
const ITensorInfo * |
gamma = nullptr, |
|
|
const ITensorInfo * |
beta = nullptr, |
|
|
float |
epsilon = 1e-12f |
|
) |
| |
|
static |
Static function to check if given info will lead to a valid configuration of CLInstanceNormalizationLayerEx.
- Parameters
-
| [in] | input | Source tensor info. In case of output tensor = nullptr this tensor will store the result of the normalization. Data types supported: F16/F32. Data layout supported: NHWC, NCHW |
| [in] | output | Destination tensor info. Data types and data layouts supported: same as input. |
| [in] | gamma | (Optional) The scale tensor applied to the normalized tensor. Defaults to nullptr |
| [in] | beta | (Optional) The offset tensor applied to the normalized tensor. Defaults to nullptr |
| [in] | epsilon | (Optional) Lower bound value for the normalization. Defaults to 1e-12 |
- Returns
- a status
The documentation for this class was generated from the following file: