21 m_Weight = std::make_unique<ScopedCpuTensorHandle>(*(descriptor.
m_Weight));
26 m_Bias = std::make_unique<ScopedCpuTensorHandle>(*(descriptor.
m_Bias));
35 const TensorInfo& weightsInfo = m_Weight->GetTensorInfo();
38 const int16_t* inputData = GetInputTensorData<int16_t>(0,
m_Data);;
39 const int16_t* weightsData = m_Weight->template GetConstTensor<int16_t>();
42 EthosnRefTransposeConvolutionImpl<armnn::TransposeConvolution2dQueueDescriptor, int16_t, int16_t, int32_t, int64_t>(
47 outputInfo.GetQuantizationScale(), outputInfo.GetQuantizationOffset(), weightsInfo);
50 const int8_t* inputData = GetInputTensorData<int8_t>(0,
m_Data);;
51 const int8_t* weightsData = m_Weight->template GetConstTensor<int8_t>();
54 EthosnRefTransposeConvolutionImpl<armnn::TransposeConvolution2dQueueDescriptor, int8_t, int8_t, int32_t, int32_t>(
59 outputInfo.GetQuantizationScale(), outputInfo.GetQuantizationOffset(), weightsInfo);
63 const uint8_t* inputData = GetInputTensorData<uint8_t>(0,
m_Data);;
64 const uint8_t* weightsData = m_Weight->template GetConstTensor<uint8_t>();
67 EthosnRefTransposeConvolutionImpl<armnn::TransposeConvolution2dQueueDescriptor, uint8_t, uint8_t, int32_t, int32_t>(
72 outputInfo.GetQuantizationScale(), outputInfo.GetQuantizationOffset(), weightsInfo);
bool m_BiasEnabled
Enable/disable bias.
EthosnRefTransposeConvolution2dWorkload(const TransposeConvolution2dQueueDescriptor &descriptor, const WorkloadInfo &info)
void Execute() const override
Copyright (c) 2021 ARM Limited and Contributors.
LayerDescriptor m_Parameters
bool CheckDataType(DataType type, DataType inputType, DataType weightsType)
int32_t GetQuantizationOffset() const
float GetQuantizationScale() const
TransposeConvolution2dQueueDescriptor m_Data
DataType GetDataType() const
#define ARMNN_SCOPED_PROFILING_EVENT_ETHOSN(name)
std::vector< ITensorHandle * > m_Outputs
Contains information about TensorInfos of a layer.
const ConstTensorHandle * m_Weight
std::vector< ITensorHandle * > m_Inputs
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
const ConstTensorHandle * m_Bias