24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_GRAPH_BATCH_NORMALIZATION_LAYER_NODE_H
25 #define ARM_COMPUTE_GRAPH_BATCH_NORMALIZATION_LAYER_NODE_H
Batch Normalization Layer node.
void accept(INodeVisitor &v) override
Accepts a node visitor.
float epsilon() const
Epsilon parameter accessor.
Activation Layer Information class.
void set_fused_activation(ActivationLayerInfo fused_activation)
Sets fused activation.
static constexpr NodeType node_type
ActivationLayerInfo fused_activation() const
Returns fused activation.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
Copyright (c) 2017-2024 Arm Limited.
NodeType type() const override
Returns node's type.
BatchNormalizationLayerNode(float epsilon=1.f, ActivationLayerInfo fused_activation=ActivationLayerInfo())
Constructor.
@ BatchNormalizationLayer