24 #ifndef ARM_COMPUTE_GRAPH_FUSED_DEPTHWISE_CONVOLUTION_BATCH_NORMALIZATION_NODE_H 25 #define ARM_COMPUTE_GRAPH_FUSED_DEPTHWISE_CONVOLUTION_BATCH_NORMALIZATION_NODE_H 96 int depth_multiplier);
129 unsigned int _depth_multiplier;
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
Fused Depthwise Convolution Batch Normalization node.
void set_convolution_method(ConvolutionMethod method)
Sets the convolution layer method to use.
PadStrideInfo convolution_info() const
Convolution metadata accessor.
NodeType type() const override
Returns node's type.
unsigned int depth_multiplier() const
Depth multiplier accessor.
DepthwiseConvolutionMethod depthwise_convolution_method() const
Depthwise convolution layer method accessor.
void set_depthwise_convolution_method(DepthwiseConvolutionMethod method)
Sets the depthwise convolution layer method to use.
Activation Layer Information class.
FusedDepthwiseConvolutionBatchNormalizationNode(float epsilon, PadStrideInfo info, unsigned int depth_multiplier, DepthwiseConvolutionMethod method, ActivationLayerInfo fused_activation=ActivationLayerInfo())
Constructor.
Copyright (c) 2017-2021 Arm Limited.
void accept(INodeVisitor &v) override
Accepts a node visitor.
void set_fused_activation(ActivationLayerInfo fused_activation)
Sets fused activation.
static constexpr NodeType node_type
ActivationLayerInfo fused_activation() const
Returns fused activation.
DepthwiseConvolutionMethod
Supported Depthwise Convolution layer methods.
static TensorDescriptor compute_output_descriptor(const TensorDescriptor &input_descriptor, const TensorDescriptor &weights_descriptor, const PadStrideInfo &info, int depth_multiplier)
Computes convolution output descriptor.
Padding and stride information class.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
ConvolutionMethod
Supported Convolution layer methods.
float epsilon() const
Epsilon parameter accessor.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.