23.02
|
Go to the documentation of this file.
51 std::vector<ITensorHandle*> outputs)
const
59 std::unique_ptr<Decoder<float>> inputDecoder = MakeDecoder<float>(
GetTensorInfo(inputs[0]), inputs[0]->
Map());
60 std::unique_ptr<Encoder<float>> outputEncoder = MakeEncoder<float>(
GetTensorInfo(outputs[0]), outputs[0]->
Map());
61 std::unique_ptr<Decoder<float>> filterDecoder = MakeDecoder<float>(
GetTensorInfo(inputs[1]), inputs[1]->
Map());
62 std::unique_ptr<Decoder<float>> biasDecoder{};
65 biasDecoder = MakeDecoder<float>(
GetTensorInfo(inputs[2]), inputs[2]->Map());
68 Convolve(inputShape, *inputDecoder, outputShape, *outputEncoder,
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
Optional< TensorInfo > m_BiasTensorInfo
std::vector< ITensorHandle * > m_Inputs
Depthwise Convolution 2D layer workload data.
bool m_BiasEnabled
Enable/disable bias.
DepthwiseConvolution2dQueueDescriptor m_Data
Optional< TensorInfo > m_WeightsTensorInfo
uint32_t m_StrideX
Stride value when proceeding through input for the width dimension.
uint32_t m_DilationX
Dilation factor value for width dimension.
uint32_t m_PadLeft
Padding left value in the width dimension.
Copyright (c) 2021 ARM Limited and Contributors.
uint32_t m_PadTop
Padding top value in the height dimension.
uint32_t m_DilationY
Dilation factor value for height dimension.
std::vector< TensorInfo > m_OutputTensorInfos
DataLayout m_DataLayout
The data layout to be used (NCHW, NHWC).
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
RefDepthwiseConvolution2dWorkload(const DepthwiseConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info)
void Execute() const override
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)
const TensorShape & GetShape() const
Contains information about TensorInfos of a layer.
LayerDescriptor m_Parameters
#define ARMNN_REPORT_PROFILING_WORKLOAD_DESC(name, desc, infos, guid)
std::vector< ITensorHandle * > m_Outputs
std::vector< ITensorHandle * > m_Outputs
uint32_t m_StrideY
Stride value when proceeding through input for the height dimension.
void ExecuteAsync(ExecutionData &executionData) override
@ CpuRef
CPU Execution: Reference C++ kernels.
std::vector< TensorInfo > m_InputTensorInfos
std::vector< ITensorHandle * > m_Inputs