21.02
|
Batch Normalization node. More...
#include <FusedConvolutionBatchNormalizationNode.h>
Public Member Functions | |
FusedConvolutionBatchNormalizationNode (float epsilon, PadStrideInfo info, unsigned int num_groups=1, ConvolutionMethod method=ConvolutionMethod::Default, FastMathHint fast_math_hint=FastMathHint::Disabled, ActivationLayerInfo fused_activation=ActivationLayerInfo()) | |
Constructor. More... | |
float | epsilon () const |
Epsilon parameter accessor. More... | |
ActivationLayerInfo | fused_activation () const |
Returns fused activation. More... | |
void | set_fused_activation (ActivationLayerInfo fused_activation) |
Sets fused activation. More... | |
void | set_convolution_method (ConvolutionMethod method) |
Sets the convolution layer method to use. More... | |
unsigned int | num_groups () const |
Number of groups in convolution accessor. More... | |
ConvolutionMethod | convolution_method () const |
Convolution layer method accessor. More... | |
void | set_fast_math_hint (FastMathHint hint) |
Sets the fast math fast hint. More... | |
FastMathHint | fast_math_hint () const |
Fast math hint accessor. More... | |
PadStrideInfo | convolution_info () const |
Convolution metadata accessor. More... | |
NodeType | type () const override |
Returns node's type. More... | |
bool | forward_descriptors () override |
Forwards descriptor information to outputs if possible. More... | |
TensorDescriptor | configure_output (size_t idx) const override |
Calculates output configuration. More... | |
void | accept (INodeVisitor &v) override |
Accepts a node visitor. More... | |
Public Member Functions inherited from INode | |
INode () | |
Constructor. More... | |
virtual | ~INode ()=default |
Destructor. More... | |
INode (const INode &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
INode & | operator= (const INode &)=delete |
Prevent instances of this class from being copy assigned (As this class contains pointers) More... | |
INode (INode &&)=default | |
Allow instances of this class to be moved. More... | |
INode & | operator= (INode &&)=default |
Allow instances of this class to be move assigned. More... | |
virtual Status | validate () const |
Validate node. More... | |
std::string | name () const |
Returns node's name. More... | |
NodeID | id () const |
Returns node's ID. More... | |
const Graph * | graph () const |
Returns node's Graph. More... | |
Graph * | graph () |
Returns node's Graph. More... | |
void | set_graph (Graph *g) |
Sets the graph that this node is registered to. More... | |
void | set_id (NodeID id) |
Sets the node id. More... | |
void | set_common_node_parameters (NodeParams common_params) |
Sets common node parameters. More... | |
void | set_requested_target (Target target) |
Sets target preference. More... | |
void | set_assigned_target (Target target) |
Sets the final execution target. More... | |
void | set_output_tensor (TensorID tid, size_t idx) |
Sets the output tensor of at a given index. More... | |
const std::vector< TensorID > & | inputs () const |
Returns inputs of the node. More... | |
const std::vector< TensorID > & | outputs () const |
Returns outputs of the node. More... | |
const std::vector< EdgeID > & | input_edges () const |
Returns input edge set. More... | |
const std::set< EdgeID > & | output_edges () const |
Returns output edge set. More... | |
TensorID | input_id (size_t idx) const |
Returns the tensor ID of a given input of the node. More... | |
TensorID | output_id (size_t idx) const |
Returns the tensor ID of a given output of the node. More... | |
Tensor * | input (size_t idx) const |
Returns the tensor of a given input of the node. More... | |
Tensor * | output (size_t idx) const |
Returns the tensor of a given output of the node. More... | |
EdgeID | input_edge_id (size_t idx) const |
Returns the edge ID of a given input of the node. More... | |
Edge * | input_edge (size_t idx) const |
Returns the edge of a given input of the node. More... | |
size_t | num_inputs () const |
Returns number of inputs of the node. More... | |
size_t | num_outputs () const |
Returns number of outputs of the node. More... | |
NodeParams | common_node_params () const |
Returns common node parameters. More... | |
Target | requested_target () const |
Returns requested target for this node. More... | |
Target | assigned_target () const |
Returns assigned target for this node. More... | |
Static Public Member Functions | |
static TensorDescriptor | compute_output_descriptor (const TensorDescriptor &input_descriptor, const TensorDescriptor &weights_descriptor, const PadStrideInfo &info) |
Computes convolution output descriptor. More... | |
Static Public Attributes | |
static constexpr NodeType | node_type = NodeType::FusedConvolutionBatchNormalizationLayer |
Batch Normalization node.
Definition at line 34 of file FusedConvolutionBatchNormalizationNode.h.
FusedConvolutionBatchNormalizationNode | ( | float | epsilon, |
PadStrideInfo | info, | ||
unsigned int | num_groups = 1 , |
||
ConvolutionMethod | method = ConvolutionMethod::Default , |
||
FastMathHint | fast_math_hint = FastMathHint::Disabled , |
||
ActivationLayerInfo | fused_activation = ActivationLayerInfo() |
||
) |
Constructor.
[in] | epsilon | Epsilon parameter. |
[in] | info | Convolution layer attributes. |
[in] | num_groups | (Optional) Number of groups (Defaults to 1) |
[in] | method | (Optional) Convolution method to use |
[in] | fast_math_hint | (Optional) Fast math hint |
[in] | fused_activation | (Optional) Fused activation layer. Disabled if not specified |
Definition at line 35 of file FusedConvolutionBatchNormalizationNode.cpp.
|
overridevirtual |
Accepts a node visitor.
[in] | v | Visitor to accept |
Implements INode.
Definition at line 144 of file FusedConvolutionBatchNormalizationNode.cpp.
References INodeVisitor::visit().
|
static |
Computes convolution output descriptor.
[in] | input_descriptor | Input descriptor |
[in] | weights_descriptor | Weights descriptor |
[in] | info | Convolution operation attributes |
Definition at line 91 of file FusedConvolutionBatchNormalizationNode.cpp.
References arm_compute::CHANNEL, arm_compute::test::validation::data_layout, arm_compute::graph::get_dimension_idx(), arm_compute::graph::get_dimension_size(), arm_compute::HEIGHT, input_height, input_width, TensorDescriptor::layout, arm_compute::scaled_dimensions(), TensorShape::set(), TensorDescriptor::shape, and arm_compute::WIDTH.
Referenced by FusedConvolutionBatchNormalizationNode::configure_output().
|
overridevirtual |
Calculates output configuration.
[in] | idx | Output index to configure |
Implements INode.
Definition at line 126 of file FusedConvolutionBatchNormalizationNode.cpp.
References ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_UNUSED, FusedConvolutionBatchNormalizationNode::compute_output_descriptor(), Tensor::desc(), INode::input(), arm_compute::test::validation::output_info, and arm_compute::test::validation::src.
Referenced by FusedConvolutionBatchNormalizationNode::forward_descriptors().
PadStrideInfo convolution_info | ( | ) | const |
Convolution metadata accessor.
Definition at line 71 of file FusedConvolutionBatchNormalizationNode.cpp.
ConvolutionMethod convolution_method | ( | ) | const |
Convolution layer method accessor.
Definition at line 56 of file FusedConvolutionBatchNormalizationNode.cpp.
float epsilon | ( | ) | const |
Epsilon parameter accessor.
Definition at line 51 of file FusedConvolutionBatchNormalizationNode.cpp.
FastMathHint fast_math_hint | ( | ) | const |
Fast math hint accessor.
Definition at line 66 of file FusedConvolutionBatchNormalizationNode.cpp.
|
overridevirtual |
Forwards descriptor information to outputs if possible.
Implements INode.
Definition at line 114 of file FusedConvolutionBatchNormalizationNode.cpp.
References ARM_COMPUTE_ERROR_ON, FusedConvolutionBatchNormalizationNode::configure_output(), Tensor::desc(), arm_compute::test::validation::dst, INode::input_id(), arm_compute::graph::NullTensorID, INode::output(), and INode::output_id().
ActivationLayerInfo fused_activation | ( | ) | const |
Returns fused activation.
Definition at line 81 of file FusedConvolutionBatchNormalizationNode.cpp.
Referenced by FusedConvolutionBatchNormalizationNode::set_fused_activation().
unsigned int num_groups | ( | ) | const |
Number of groups in convolution accessor.
Definition at line 76 of file FusedConvolutionBatchNormalizationNode.cpp.
void set_convolution_method | ( | ConvolutionMethod | method | ) |
Sets the convolution layer method to use.
[in] | method | Method to use for convolution |
Definition at line 46 of file FusedConvolutionBatchNormalizationNode.cpp.
void set_fast_math_hint | ( | FastMathHint | hint | ) |
Sets the fast math fast hint.
[in] | hint | Hint to use for convolution |
Definition at line 61 of file FusedConvolutionBatchNormalizationNode.cpp.
void set_fused_activation | ( | ActivationLayerInfo | fused_activation | ) |
Sets fused activation.
[in] | fused_activation | Fused activation to set |
Definition at line 86 of file FusedConvolutionBatchNormalizationNode.cpp.
References FusedConvolutionBatchNormalizationNode::fused_activation().
|
overridevirtual |
Returns node's type.
Implements INode.
Definition at line 139 of file FusedConvolutionBatchNormalizationNode.cpp.
References FusedConvolutionBatchNormalizationNode::node_type.
|
static |
Definition at line 128 of file FusedConvolutionBatchNormalizationNode.h.
Referenced by FusedConvolutionBatchNormalizationNode::type().