22 m_Weight = std::make_unique<ScopedCpuTensorHandle>(*(descriptor.
m_Weight));
23 const TensorInfo& rFilterInfo = m_Weight->GetTensorInfo();
24 m_FilterShape = rFilterInfo.
GetShape();
25 m_FilterDecoder = MakeDecoder<float>(rFilterInfo, m_Weight->Map(
true));
29 m_Bias = std::make_unique<ScopedCpuTensorHandle>(*(descriptor.
m_Bias));
30 const TensorInfo& biasInfo = m_Bias->GetTensorInfo();
31 m_BiasDecoder = MakeDecoder<float>(biasInfo, m_Bias->Map(
true));
39 m_InputDecoder = MakeDecoder<float>(inputInfo);
42 m_OutputShape = outputInfo.GetShape();
43 m_OutputEncoder = MakeEncoder<float>(outputInfo);
49 std::unique_ptr<Decoder<float>> pBiasDecoder{};
54 Convolve(m_InputShape, *m_InputDecoder, m_OutputShape, *m_OutputEncoder,
bool m_BiasEnabled
Enable/disable bias.
const TensorShape & GetShape() const
CPU Execution: Reference C++ kernels.
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
const ConstCpuTensorHandle * m_Bias
uint32_t m_PadLeft
Padding left value in the width dimension.
const DepthwiseConvolution2dQueueDescriptor m_Data
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
Copyright (c) 2020 ARM Limited.
void PostAllocationConfigure() override
LayerDescriptor m_Parameters
uint32_t m_DilationY
Dilation factor value for height dimension.
const ConstCpuTensorHandle * m_Weight
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
uint32_t m_StrideX
Stride value when proceeding through input for the width dimension.
void Convolve(const TensorShape &rInputShape, Decoder< float > &rInputDecoder, const TensorShape &rOutputShape, Encoder< float > &rOutputEncoder, const TensorShape &rFilterShape, Decoder< float > &rFilterDecoder, bool biasEnabled, Decoder< float > *pBiasDecoder, DataLayout dataLayout, unsigned int paddingTop, unsigned int paddingLeft, unsigned int xStride, unsigned int yStride, unsigned int xDilation, unsigned int yDilation, bool depthwise)
uint32_t m_DilationX
Dilation factor value for width dimension.
uint32_t m_PadTop
Padding top value in the height dimension.
virtual void Execute() const override
RefDepthwiseConvolution2dWorkload(const DepthwiseConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info)
uint32_t m_StrideY
Stride value when proceeding through input for the height dimension.
std::vector< ITensorHandle * > m_Outputs
Contains information about inputs and outputs to a layer.
std::vector< ITensorHandle * > m_Inputs