ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Public Member Functions | Static Public Member Functions
arm_compute::NEInstanceNormalizationLayerKernelEx Class Reference

#include <NEInstanceNormalizationLayerKernelEx.h>

Collaboration diagram for arm_compute::NEInstanceNormalizationLayerKernelEx:
Collaboration graph
[legend]

Public Member Functions

const char * name () const override
 
 NEInstanceNormalizationLayerKernelEx ()
 
 NEInstanceNormalizationLayerKernelEx (const NEInstanceNormalizationLayerKernelEx &)=delete
 
NEInstanceNormalizationLayerKernelEx & operator= (const NEInstanceNormalizationLayerKernelEx &)=delete
 
 NEInstanceNormalizationLayerKernelEx (NEInstanceNormalizationLayerKernelEx &&)=default
 
NEInstanceNormalizationLayerKernelEx & operator= (NEInstanceNormalizationLayerKernelEx &&)=default
 
 ~NEInstanceNormalizationLayerKernelEx ()=default
 
void configure (ITensor *input, ITensor *output, ITensor *gamma=nullptr, ITensor *beta=nullptr, float epsilon=1e-12f)
 
void run (const Window &window, const ThreadInfo &info) 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 NEInstanceNormalizationLayerKernelEx.h.

Constructor & Destructor Documentation

◆ NEInstanceNormalizationLayerKernelEx() [1/3]

arm_compute::NEInstanceNormalizationLayerKernelEx::NEInstanceNormalizationLayerKernelEx ( )

Default constructor

◆ NEInstanceNormalizationLayerKernelEx() [2/3]

arm_compute::NEInstanceNormalizationLayerKernelEx::NEInstanceNormalizationLayerKernelEx ( const NEInstanceNormalizationLayerKernelEx &  )
delete

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

◆ NEInstanceNormalizationLayerKernelEx() [3/3]

arm_compute::NEInstanceNormalizationLayerKernelEx::NEInstanceNormalizationLayerKernelEx ( NEInstanceNormalizationLayerKernelEx &&  )
default

Allow instances of this class to be moved

◆ ~NEInstanceNormalizationLayerKernelEx()

arm_compute::NEInstanceNormalizationLayerKernelEx::~NEInstanceNormalizationLayerKernelEx ( )
default

Default destructor

Member Function Documentation

◆ configure()

void arm_compute::NEInstanceNormalizationLayerKernelEx::configure ( ITensor *  input,
ITensor *  output,
ITensor *  gamma = nullptr,
ITensor *  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 In case of output tensor = nullptr this tensor will store the result of the normalization.
[out]outputDestination tensor. Data types and data layouts supported: same as input.
[in]gamma(Optional) The scale scalar value applied to the normalized tensor. Defaults to 1.0
[in]beta(Optional) The offset scalar value applied to the normalized tensor. Defaults to 0.0
[in]epsilon(Optional) Lower bound value for the normalization. Defaults to 1e-12

◆ name()

const char * arm_compute::NEInstanceNormalizationLayerKernelEx::name ( ) const
inlineoverride

Definition at line 54 of file NEInstanceNormalizationLayerKernelEx.h.

54{ return "NEInstanceNormalizationLayerKernelEx"; }

◆ operator=() [1/2]

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

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

◆ operator=() [2/2]

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

Allow instances of this class to be moved

◆ run()

void arm_compute::NEInstanceNormalizationLayerKernelEx::run ( const Window &  window,
const ThreadInfo &  info 
)
override

◆ validate()

static Status arm_compute::NEInstanceNormalizationLayerKernelEx::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 NEInstanceNormalizationLayer.

Parameters
[in]inputSource tensor info. Data types supported: F16/F32. Data layout supported: NCHW
[in]outputDestination tensor info. Data types and data layouts supported: same as input.
[in]gamma(Optional) The scale scalar value applied to the normalized tensor. Defaults to 1.0
[in]beta(Optional) The offset scalar value applied to the normalized tensor. Defaults to 0.0
[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: