24.02.1
|
Go to the documentation of this file.
35 : _epsilon(
epsilon), _fused_activation(fused_activation)
48 return _fused_activation;
constexpr EdgeID EmptyEdgeID
SimpleTensor< float > src
TensorID input_id(size_t idx) const
Returns the tensor ID of a given input of the node.
Tensor * output(size_t idx) const
Returns the tensor of a given output of the node.
void accept(INodeVisitor &v) override
Accepts a node visitor.
float epsilon() const
Epsilon parameter accessor.
Activation Layer Information class.
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
void set_fused_activation(ActivationLayerInfo fused_activation)
Sets fused activation.
virtual void visit(INode &n)=0
Visit INode.
static constexpr NodeType node_type
ActivationLayerInfo fused_activation() const
Returns fused activation.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
Copyright (c) 2017-2024 Arm Limited.
Tensor * input(size_t idx) const
Returns the tensor of a given input of the node.
TensorID output_id(size_t idx) const
Returns the tensor ID of a given output of the node.
NodeType type() const override
Returns node's type.
constexpr TensorID NullTensorID
Constant NodeID specifying an equivalent of null node.
BatchNormalizationLayerNode(float epsilon=1.f, ActivationLayerInfo fused_activation=ActivationLayerInfo())
Constructor.