21.02
|
Batchnormalization Layer. More...
#include <Layers.h>
Public Member Functions | |
BatchNormalizationLayer (ITensorAccessorUPtr mean, ITensorAccessorUPtr var, ITensorAccessorUPtr gamma=nullptr, ITensorAccessorUPtr beta=nullptr, float epsilon=0.001f) | |
Construct a batch normalization layer. More... | |
NodeID | create_layer (IStream &s) override |
Create layer and add to the given stream. More... | |
Public Member Functions inherited from ILayer | |
virtual | ~ILayer ()=default |
Default destructor. More... | |
ILayer & | set_name (std::string name) |
Sets the name of the layer. More... | |
const std::string & | name () const |
Layer name accessor. More... | |
|
inline |
Construct a batch normalization layer.
[in] | mean | Accessor to get mean tensor data from. |
[in] | var | Accessor to get var tensor data from. |
[in] | gamma | (Optional) Accessor to get gamma tensor data from. Default: nullptr. |
[in] | beta | (Optional) Accessor to get beta tensor data from. Default: nullptr. |
[in] | epsilon | (Optional) Epsilon value. Default: 0.001. |
Definition at line 202 of file Layers.h.
Create layer and add to the given stream.
[in] | s | Stream to add layer to. |
Implements ILayer.
Definition at line 211 of file Layers.h.
References GraphBuilder::add_batch_normalization_node(), ARM_COMPUTE_ERROR_ON, IStream::graph(), IStream::hints(), arm_compute::test::validation::input, ILayer::name(), IStream::tail_node(), and StreamHints::target_hint.