41 std::vector<ITensorHandle*> outputs)
45 m_FilterShape = rFilterInfo.
GetShape();
46 m_FilterDecoder = MakeDecoder<float>(rFilterInfo);
51 m_BiasDecoder = MakeDecoder<float>(biasInfo);
71 std::unique_ptr<Decoder<float>> inputDecoder = MakeDecoder<float>(
GetTensorInfo(inputs[0]), inputs[0]->Map());
72 std::unique_ptr<Encoder<float>> outputEncoder = MakeEncoder<float>(
GetTensorInfo(outputs[0]), outputs[0]->Map());
77 m_FilterDecoder->Reset(inputs[1]->
Map());
80 m_BiasDecoder->Reset(inputs[2]->
Map());
83 Convolve3d(inputShape, *inputDecoder, outputShape, *outputEncoder, m_FilterShape,
const TensorShape & GetShape() const
CPU Execution: Reference C++ kernels.
void Execute() const override
uint32_t m_StrideX
Stride value when proceeding through input for the width dimension.
bool m_BiasEnabled
Enable/disable bias.
Copyright (c) 2021 ARM Limited and Contributors.
void IgnoreUnused(Ts &&...)
LayerDescriptor m_Parameters
uint32_t m_DilationX
Dilation along x axis.
uint32_t m_StrideY
Stride value when proceeding through input for the height dimension.
std::vector< ITensorHandle * > m_Inputs
void PostAllocationConfigure() override
std::vector< TensorInfo > m_InputTensorInfos
Convolution3dQueueDescriptor m_Data
void ExecuteAsync(WorkingMemDescriptor &workingMemDescriptor) override
std::vector< TensorInfo > m_OutputTensorInfos
uint32_t m_PadFront
Padding front value in the depth dimension.
uint32_t m_PadLeft
Padding left value in the width dimension.
#define ARMNN_SCOPED_PROFILING_EVENT_GUID(backendId, name, guid)
profiling::ProfilingGuid GetGuid() const final
Optional< TensorInfo > m_BiasTensorInfo
DataLayout m_DataLayout
The data layout to be used (NDHWC, NCDHW).
std::vector< ITensorHandle * > m_Outputs
uint32_t m_PadTop
Padding top value in the height dimension.
std::vector< ITensorHandle * > m_Outputs
#define ARMNN_REPORT_PROFILING_WORKLOAD_DESC(name, desc, infos, guid)
Contains information about TensorInfos of a layer.
std::vector< ITensorHandle * > m_Inputs
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
uint32_t m_DilationZ
Dilation along z axis.
RefConvolution3dWorkload(const Convolution3dQueueDescriptor &descriptor, const WorkloadInfo &info)
uint32_t m_StrideZ
Stride value when proceeding through input for the depth dimension.
Optional< TensorInfo > m_WeightsTensorInfo
uint32_t m_DilationY
Dilation along y axis.
void Convolve3d(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 paddingFront, unsigned int xStride, unsigned int yStride, unsigned int zStride, unsigned int xDilation, unsigned int yDilation, unsigned int zDilation)