24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_GRAPH_FUSED_CONVOLUTION_BATCH_NORMALIZATION_NODE_H
25 #define ARM_COMPUTE_GRAPH_FUSED_CONVOLUTION_BATCH_NORMALIZATION_NODE_H
135 unsigned int _num_groups;
void set_fast_math_hint(FastMathHint hint)
Sets the fast math hint.
@ Default
Default approach using internal heuristics.
unsigned int num_groups() const
Number of groups in convolution accessor.
void set_fused_activation(ActivationLayerInfo fused_activation)
Sets fused activation.
PadStrideInfo convolution_info() const
Convolution metadata accessor.
Activation Layer Information class.
float epsilon() const
Epsilon parameter accessor.
ActivationLayerInfo fused_activation() const
Returns fused activation.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
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.
static TensorDescriptor compute_output_descriptor(const TensorDescriptor &input_descriptor, const TensorDescriptor &weights_descriptor, const PadStrideInfo &info)
Computes convolution output descriptor.
void set_convolution_method(ConvolutionMethod method)
Sets the convolution layer method to use.
void accept(INodeVisitor &v) override
Accepts a node visitor.
Batch Normalization node.
@ Disabled
Fast math disabled for Convolution layer.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
ConvolutionMethod convolution_method() const
Convolution layer method accessor.
FastMathHint
Enable or disable fast math for Convolution layer.
NodeType type() const override
Returns node's type.
ConvolutionMethod
Supported Convolution layer methods.
Copyright (c) 2017-2024 Arm Limited.
@ FusedConvolutionBatchNormalizationLayer
static constexpr NodeType node_type
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
FastMathHint fast_math_hint() const
Fast math hint accessor.