Compute Library
DeconvolutionLayer Class Referencefinal

Deconvolution Layer. More...

#include <Layers.h>

Collaboration diagram for DeconvolutionLayer:

Public Member Functions

 DeconvolutionLayer (unsigned int conv_width, unsigned int conv_height, unsigned int ofm, ITensorAccessorUPtr weights, ITensorAccessorUPtr bias, PadStrideInfo deconv_info)
 Construct a convolution layer. More...
NodeID create_layer (IStream &s) override
 Create layer and add to the given stream. More...
- Public Member Functions inherited from ILayer
virtual ~ILayer ()=default
 Default destructor. More...
ILayerset_name (std::string name)
 Sets the name of the layer. More...
const std::string & name () const
 Layer name accessor. More...

Detailed Description

Deconvolution Layer.

Definition at line 435 of file Layers.h.

Constructor & Destructor Documentation

◆ DeconvolutionLayer()

DeconvolutionLayer ( unsigned int  conv_width,
unsigned int  conv_height,
unsigned int  ofm,
ITensorAccessorUPtr  weights,
ITensorAccessorUPtr  bias,
PadStrideInfo  deconv_info 

Construct a convolution layer.

[in]conv_widthConvolution width.
[in]conv_heightConvolution height.
[in]ofmOutput feature map.
[in]weightsAccessor to get kernel weights from.
[in]biasAccessor to get kernel bias from.
[in]deconv_infoPadding and stride information.

Definition at line 447 of file Layers.h.

453  : _conv_width(conv_width),
454  _conv_height(conv_height),
455  _ofm(ofm),
456  _deconv_info(std::move(deconv_info)),
457  _weights(std::move(weights)),
458  _bias(std::move(bias))
459  {
460  }
const int32_t * bias

Member Function Documentation

◆ create_layer()

NodeID create_layer ( IStream s)

Create layer and add to the given stream.

[in]sStream to add layer to.
ID of the created node.

Implements ILayer.

Definition at line 462 of file Layers.h.

References GraphBuilder::add_deconvolution_node(), IStream::graph(), IStream::hints(), arm_compute::test::validation::input, ILayer::name(), IStream::tail_node(), and StreamHints::target_hint.

463  {
464  NodeIdxPair input = { s.tail_node(), 0 };
465  NodeParams common_params = { name(), s.hints().target_hint };
466  return GraphBuilder::add_deconvolution_node(s.graph(), common_params, input,
467  Size2D(_conv_width, _conv_height), _ofm, _deconv_info,
468  std::move(_weights), std::move(_bias));
469  }
static NodeID add_deconvolution_node(Graph &g, NodeParams params, NodeIdxPair input, Size2D kernel_spatial_extend, unsigned int depth, PadStrideInfo deconv_info, ITensorAccessorUPtr weights_accessor=nullptr, ITensorAccessorUPtr bias_accessor=nullptr)
Adds a deconvolution layer node to the graph.
const std::string & name() const
Layer name accessor.
Definition: ILayer.h:64

The documentation for this class was generated from the following file: