21 m_Weight = std::make_unique<ScopedCpuTensorHandle>(*(descriptor.
m_Weight));
25 m_Bias = std::make_unique<ScopedCpuTensorHandle>(*(descriptor.
m_Bias));
33 const TensorInfo& weightsInfo = m_Weight->GetTensorInfo();
36 const int16_t* inputData = GetInputTensorData<int16_t>(0,
m_Data);;
37 const int16_t* weightsData = m_Weight->template GetConstTensor<int16_t>();
40 const TensorInfo& filterInfo = m_Weight->GetTensorInfo();
41 EthosnRefConvImpl<armnn::DepthwiseConvolution2dQueueDescriptor, int16_t, int16_t, int32_t, int64_t>(
46 outputInfo.GetQuantizationScale(), outputInfo.GetQuantizationOffset(), filterInfo,
true);
49 const int8_t* inputData = GetInputTensorData<int8_t>(0,
m_Data);;
50 const int8_t* weightsData = m_Weight->template GetConstTensor<int8_t>();
53 const TensorInfo& filterInfo = m_Weight->GetTensorInfo();
54 EthosnRefConvImpl<armnn::DepthwiseConvolution2dQueueDescriptor, int8_t, int8_t, int32_t, int64_t>(
59 outputInfo.GetQuantizationScale(), outputInfo.GetQuantizationOffset(), filterInfo,
true);
64 const uint8_t* inputData = GetInputTensorData<uint8_t>(0,
m_Data);;
65 const uint8_t* weightsData = m_Weight->template GetConstTensor<uint8_t>();
68 const TensorInfo& filterInfo = m_Weight->GetTensorInfo();
69 EthosnRefConvImpl<armnn::DepthwiseConvolution2dQueueDescriptor, uint8_t, uint8_t, int32_t, int32_t>(
74 outputInfo.GetQuantizationScale(), outputInfo.GetQuantizationOffset(), filterInfo,
true);
bool m_BiasEnabled
Enable/disable bias.
Copyright (c) 2021 ARM Limited and Contributors.
LayerDescriptor m_Parameters
virtual void Execute() const override
bool CheckDataType(DataType type, DataType inputType, DataType weightsType)
int32_t GetQuantizationOffset() const
float GetQuantizationScale() const
DepthwiseConvolution2dQueueDescriptor m_Data
DataType GetDataType() const
const ConstTensorHandle * m_Bias
const ConstTensorHandle * m_Weight
#define ARMNN_SCOPED_PROFILING_EVENT_ETHOSN(name)
std::vector< ITensorHandle * > m_Outputs
Contains information about TensorInfos of a layer.
std::vector< ITensorHandle * > m_Inputs
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
EthosnRefDepthwiseConvolution2dWorkload(const DepthwiseConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info)
Depthwise Convolution 2D layer workload data.