24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_GRAPH_DEPTHWISE_CONVOLUTION_LAYER_NODE_H
25 #define ARM_COMPUTE_GRAPH_DEPTHWISE_CONVOLUTION_LAYER_NODE_H
111 int _depth_multiplier;
void accept(INodeVisitor &v) override
Accepts a node visitor.
Quantization information.
static constexpr NodeType node_type
void set_fused_activation(ActivationLayerInfo fused_activation)
Sets fused activation.
ActivationLayerInfo fused_activation() const
Returns fused activation.
static TensorDescriptor compute_output_descriptor(const TensorDescriptor &input_descriptor, const TensorDescriptor &weights_descriptor, const PadStrideInfo &info, int depth_multiplier=1)
Computes depthwise convolution output descriptor.
DepthwiseConvolutionMethod depthwise_convolution_method() const
Depthwise convolution layer method accessor.
Activation Layer Information class.
void set_convolution_info(PadStrideInfo info)
Sets convolution info.
TensorDescriptor configure_output(size_t idx) const override
Calculates output configuration.
bool forward_descriptors() override
Forwards descriptor information to outputs if possible.
void set_depthwise_convolution_method(DepthwiseConvolutionMethod method)
Sets the depthwise convolution method to use.
NodeType type() const override
Returns node's type.
Copyright (c) 2017-2024 Arm Limited.
DepthwiseConvolutionMethod
Supported Depthwise Convolution layer methods.
@ Default
Default approach using internal heuristics.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
DepthwiseConvolutionLayerNode(PadStrideInfo info, int depth_multiplier=1, DepthwiseConvolutionMethod method=DepthwiseConvolutionMethod::Default, QuantizationInfo out_quant_info=QuantizationInfo())
Constructor.
int depth_multiplier() const
Depth multiplier accessor.
@ DepthwiseConvolutionLayer
PadStrideInfo convolution_info() const
Convolution metadata accessor.
Depthwise Convolution Layer node.