ONE - On-device Neural Engine
Loading...
Searching...
No Matches
arm_compute::CLInstanceNormalizationLayerKernelEx Class Reference

#include <CLInstanceNormalizationLayerKernelEx.h>

Collaboration diagram for arm_compute::CLInstanceNormalizationLayerKernelEx:

Public Member Functions

 CLInstanceNormalizationLayerKernelEx ()
 
 CLInstanceNormalizationLayerKernelEx (const CLInstanceNormalizationLayerKernelEx &)=delete
 
CLInstanceNormalizationLayerKernelExoperator= (const CLInstanceNormalizationLayerKernelEx &)=delete
 
 CLInstanceNormalizationLayerKernelEx (CLInstanceNormalizationLayerKernelEx &&)=default
 
CLInstanceNormalizationLayerKernelExoperator= (CLInstanceNormalizationLayerKernelEx &&)=default
 
 ~CLInstanceNormalizationLayerKernelEx ()=default
 
void configure (ICLTensor *input, ICLTensor *output, ICLTensor *gamma=nullptr, ICLTensor *beta=nullptr, float epsilon=1e-12f)
 
void run (const Window &window, cl::CommandQueue &queue) override
 

Static Public Member Functions

static Status validate (const ITensorInfo *input, const ITensorInfo *output, const ITensorInfo *gamma=nullptr, const ITensorInfo *beta=nullptr, float epsilon=1e-12f)
 

Detailed Description

Interface for performing an instance normalization

Definition at line 51 of file CLInstanceNormalizationLayerKernelEx.h.

Constructor & Destructor Documentation

◆ CLInstanceNormalizationLayerKernelEx() [1/3]

arm_compute::CLInstanceNormalizationLayerKernelEx::CLInstanceNormalizationLayerKernelEx ( )

Constructor

◆ CLInstanceNormalizationLayerKernelEx() [2/3]

arm_compute::CLInstanceNormalizationLayerKernelEx::CLInstanceNormalizationLayerKernelEx ( const CLInstanceNormalizationLayerKernelEx )
delete

Prevent instances of this class from being copied (As this class contains pointers)

◆ CLInstanceNormalizationLayerKernelEx() [3/3]

arm_compute::CLInstanceNormalizationLayerKernelEx::CLInstanceNormalizationLayerKernelEx ( CLInstanceNormalizationLayerKernelEx &&  )
default

Default Move Constructor.

◆ ~CLInstanceNormalizationLayerKernelEx()

arm_compute::CLInstanceNormalizationLayerKernelEx::~CLInstanceNormalizationLayerKernelEx ( )
default

Default destructor

Member Function Documentation

◆ 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]inputSource tensor. Data types supported: F16/F32. Data layout supported: NCHW
[out]outputDestination 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]

CLInstanceNormalizationLayerKernelEx & arm_compute::CLInstanceNormalizationLayerKernelEx::operator= ( CLInstanceNormalizationLayerKernelEx &&  )
default

Default move assignment operator

◆ operator=() [2/2]

CLInstanceNormalizationLayerKernelEx & arm_compute::CLInstanceNormalizationLayerKernelEx::operator= ( const CLInstanceNormalizationLayerKernelEx )
delete

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]inputSource 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]outputDestination 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: