21.08
|
Normalization Layer Information class. More...
#include <Types.h>
Public Member Functions | |
NormalizationLayerInfo (NormType type, uint32_t norm_size=5, float alpha=0.0001f, float beta=0.5f, float kappa=1.f, bool is_scaled=true) | |
Default Constructor. More... | |
NormType | type () const |
Get the normalization type. More... | |
uint32_t | norm_size () const |
Get the normalization size. More... | |
float | alpha () const |
Get the alpha value. More... | |
float | beta () const |
Get the beta value. More... | |
float | kappa () const |
Get the kappa value. More... | |
bool | is_scaled () const |
Get the is_scaled value. More... | |
bool | is_cross_map () const |
Check if normalization is cross map. More... | |
bool | is_in_map () const |
Check if normalization is not cross map. More... | |
float | scale_coeff () const |
Return the scaling factor of the normalization function. More... | |
|
inline |
Default Constructor.
[in] | type | The normalization type. Can be NormType::IN_MAP_1D, NormType::IN_MAP_2D or NormType::CROSS_MAP |
[in] | norm_size | The normalization size is the number of elements to normalize across. Defaults to 5. |
[in] | alpha | (Optional) Alpha parameter used by normalization equation. Defaults to 0.0001. |
[in] | beta | (Optional) Beta parameter used by normalization equation. Defaults to 0.5. |
[in] | kappa | (Optional) Kappa parameter used by [Krichevksy 2012] Across Channel Local Brightness Normalization equation. |
[in] | is_scaled | (Optional) Boolean that specifies if alpha will be scaled by the normalization size or not. Should be false to follow [Krichevksy 2012]. |
Definition at line 1589 of file Types.h.
|
inline |
|
inline |
Get the beta value.
Definition at line 1609 of file Types.h.
Referenced by NENormalizationLayerKernel::configure(), CLNormalizationLayerKernel::configure(), and arm_compute::test::validation::reference::normalization_layer().
|
inline |
Check if normalization is cross map.
Definition at line 1624 of file Types.h.
References arm_compute::CROSS_MAP.
Referenced by CLNormalizationLayerKernel::configure(), and arm_compute::test::validation::reference::normalization_layer().
|
inline |
Check if normalization is not cross map.
Definition at line 1629 of file Types.h.
Referenced by CLNormalizationLayerKernel::configure(), and arm_compute::get_normalization_dimension_index().
|
inline |
|
inline |
Get the kappa value.
Definition at line 1614 of file Types.h.
Referenced by NENormalizationLayerKernel::configure(), CLNormalizationLayerKernel::configure(), and arm_compute::test::validation::reference::normalization_layer().
|
inline |
Get the normalization size.
Definition at line 1599 of file Types.h.
Referenced by NENormalizationLayerKernel::configure(), CLNormalizationLayerKernel::configure(), arm_compute::test::validation::reference::normalization_layer(), arm_compute::operator<<(), and arm_compute::to_string().
|
inline |
Return the scaling factor of the normalization function.
If is_scaled is set to false then [Krichevksy 2012] normalization scaling is performed, where alpha is returned plainly, else alpha is scaled by the total number of elements used for the normalization.
Definition at line 1640 of file Types.h.
References arm_compute::IN_MAP_2D.
Referenced by NENormalizationLayerKernel::configure(), CLNormalizationLayerKernel::configure(), and arm_compute::test::validation::reference::normalization_layer().
|
inline |
Get the normalization type.
Definition at line 1594 of file Types.h.
Referenced by NENormalizationLayerKernel::configure(), CLNormalizationLayerKernel::configure(), arm_compute::test::validation::reference::normalization_layer(), arm_compute::operator<<(), arm_compute::to_string(), and DotGraphVisitor::visit().