31 const std::vector<TensorShape>& inputShapes =
38 unsigned int inputChannels = filterShape[1];
39 unsigned int filterWidth = filterShape[3];
40 unsigned int filterHeight = filterShape[2];
41 unsigned int depthMultiplier = filterShape[0];
43 fn(
"FilterWidth",std::to_string(filterWidth));
44 fn(
"FilterHeight",std::to_string(filterHeight));
45 fn(
"DepthMultiplier",std::to_string(depthMultiplier));
46 fn(
"InputChannels",std::to_string(inputChannels));
73 auto layer = CloneBase<DepthwiseConvolution2dLayer>(graph,
m_Param,
GetName());
76 if (layer->m_Param.m_BiasEnabled)
81 return std::move(layer);
84 std::vector<TensorShape>
91 ARMNN_ASSERT_MSG(inputShape.GetNumDimensions() == 4,
"Convolutions will always have 4D input.");
98 unsigned int inputBatchSize = inputShape[0];
99 unsigned int inputHeight = inputShape[dataLayoutIndex.
GetHeightIndex()];
100 unsigned int inputWidth = inputShape[dataLayoutIndex.
GetWidthIndex()];
105 unsigned int filterHeight = filterShape[1];
106 unsigned int dilatedFilterHeight = filterHeight + (
m_Param.
m_DilationY - 1) * (filterHeight - 1);
110 unsigned int filterWidth = filterShape[2];
111 unsigned int dilatedFilterWidth = filterWidth + (
m_Param.
m_DilationX - 1) * (filterWidth - 1);
115 unsigned int outputChannels = filterShape[3];
116 unsigned int outputBatchSize = inputBatchSize;
119 TensorShape{ outputBatchSize, outputHeight, outputWidth, outputChannels } :
120 TensorShape{ outputBatchSize, outputChannels, outputHeight, outputWidth };
122 return std::vector<TensorShape>{ tensorShape };
138 m_Weight->GetTensorInfo().GetShape()
170 std::vector<armnn::ConstTensor> constTensors { { managedWeight.
GetTensorInfo(), managedWeight.
Map() } };
std::vector< TensorShape > InferOutputShapes(const std::vector< TensorShape > &inputShapes) const override
By default returns inputShapes if the number of inputs are equal to number of outputs, otherwise infers the output shapes from given input shapes and layer properties.
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
Makes a workload for the DepthwiseConvolution2d type.
bool m_BiasEnabled
Enable/disable bias.
DepthwiseConvolution2dDescriptor m_Param
The parameters for the layer (not including tensor-valued weights etc.).
const DepthwiseConvolution2dDescriptor & GetParameters() const
unsigned int GetWidthIndex() const
const TensorShape & GetShape() const
uint32_t m_PadBottom
Padding bottom value in the height dimension.
DepthwiseConvolution2dLayer * Clone(Graph &graph) const override
Creates a dynamically-allocated copy of this layer.
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
This layer represents a depthwise convolution 2d operation.
uint32_t m_PadLeft
Padding left value in the width dimension.
virtual void ExecuteStrategy(const armnn::IConnectableLayer *layer, const armnn::BaseDescriptor &descriptor, const std::vector< armnn::ConstTensor > &constants, const char *name, const armnn::LayerBindingId id=0)=0
ConstantTensors GetConstantTensorsByRef() override
Retrieve the handles to the constant values stored by the layer.
void SerializeLayerParameters(ParameterStringifyFunction &fn) const override
Helper to serialize the layer parameters to string (currently used in DotSerializer and company)...
void VerifyShapeInferenceType(const TensorShape &outputShape, ShapeInferenceMethod shapeInferenceMethod)
const TensorInfo & GetTensorInfo() const
Copyright (c) 2021 ARM Limited and Contributors.
uint32_t m_DilationY
Dilation factor value for height dimension.
void ValidateAndCopyShape(const TensorShape &outputShape, const TensorShape &inferredShape, const ShapeInferenceMethod shapeInferenceMethod, const std::string &layerName, const unsigned int outputSlotIndex=0)
unsigned int GetHeightIndex() const
void ValidateTensorShapesFromInputs() override
Check if the input tensor shape(s) will lead to a valid configuration of DepthwiseConvolution2dLayer...
void SerializeLayerParameters(ParameterStringifyFunction &fn) const override
Helper to serialize the layer parameters to string.
void VerifyLayerConnections(unsigned int expectedConnections, const CheckLocation &location) const
const InputSlot & GetInputSlot(unsigned int index) const override
Get a const input slot handle by slot index.
uint32_t m_StrideX
Stride value when proceeding through input for the width dimension.
std::vector< std::reference_wrapper< std::shared_ptr< ConstTensorHandle > >> ConstantTensors
uint32_t m_DilationX
Dilation factor value for width dimension.
uint32_t m_PadTop
Padding top value in the height dimension.
#define ARMNN_ASSERT_MSG(COND, MSG)
Provides access to the appropriate indexes for Channels, Height and Width based on DataLayout...
DepthwiseConvolution2dLayer(const DepthwiseConvolution2dDescriptor ¶m, const char *name)
Constructor to create a DepthwiseConvolution2dLayer.
A tensor defined by a TensorInfo (shape and data type) and an immutable backing store.
std::shared_ptr< ConstTensorHandle > m_Bias
A unique pointer to store Bias values.
const ConstTensorHandle * m_Bias
virtual void VisitDepthwiseConvolution2dLayer(const IConnectableLayer *layer, const DepthwiseConvolution2dDescriptor &convolution2dDescriptor, const ConstTensor &weights, const Optional< ConstTensor > &biases, const char *name=nullptr)=0
Function that a 2D depthwise convolution layer with biases should call back to when its Accept(ILayer...
#define ARMNN_ASSERT(COND)
const ConstTensorHandle * m_Weight
void SetAdditionalInfo(QueueDescriptor &descriptor) const
void Accept(ILayerVisitor &visitor) const override
Apply a visitor to this layer.
EmptyOptional is used to initialize the Optional class in case we want to have default value for an O...
void ExecuteStrategy(IStrategy &strategy) const override
Apply a visitor to this layer.
uint32_t m_StrideY
Stride value when proceeding through input for the height dimension.
std::shared_ptr< ConstTensorHandle > m_Weight
A unique pointer to store Weight values.
WorkloadInfo PrepInfoAndDesc(QueueDescriptor &descriptor) const
Helper function to reduce duplication in *LayerCreateWorkload.
const OutputSlot & GetOutputSlot(unsigned int index=0) const override
Get the const output slot handle by slot index.
virtual const TensorInfo & GetTensorInfo() const =0
const char * GetName() const override
Returns the name of the layer.
std::function< void(const std::string &name, const std::string &value)> ParameterStringifyFunction
virtual std::unique_ptr< IWorkload > CreateDepthwiseConvolution2d(const DepthwiseConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info) const
const void * Map(bool blocking=true)
RAII Managed resource Unmaps MemoryArea once out of scope.
const TensorInfo & GetTensorInfo() const override
ShapeInferenceMethod m_ShapeInferenceMethod
A DepthwiseConvolution2dDescriptor for the DepthwiseConvolution2dLayer.
Depthwise Convolution 2D layer workload data.
LayerType
When adding a new layer, adapt also the LastLayer enum value in the enum class LayerType below...
uint32_t m_PadRight
Padding right value in the width dimension.