20 const arm_compute::TensorInfo neonOutputInfo = armcomputetensorutils::BuildArmComputeTensorInfo(output);
22 std::array<arm_compute::DataType,8> supportedTypes = {
23 arm_compute::DataType::F16,
24 arm_compute::DataType::F32,
25 arm_compute::DataType::QASYMM8,
26 arm_compute::DataType::QASYMM8_SIGNED,
27 arm_compute::DataType::QSYMM16,
28 arm_compute::DataType::QSYMM8,
29 arm_compute::DataType::QSYMM8_PER_CHANNEL,
30 arm_compute::DataType::S32
32 auto it = std::find(begin(supportedTypes), end(supportedTypes), neonOutputInfo.data_type());
34 if (it != end(supportedTypes))
46 const arm_compute::CLCompileContext&)
68 switch (computeDataType)
70 case arm_compute::DataType::F16:
75 case arm_compute::DataType::F32:
80 case arm_compute::DataType::QASYMM8:
85 case arm_compute::DataType::QASYMM8_SIGNED:
90 case arm_compute::DataType::QSYMM16:
95 case arm_compute::DataType::QSYMM8:
96 case arm_compute::DataType::QSYMM8_PER_CHANNEL:
101 case arm_compute::DataType::S32:
#define ARMNN_SCOPED_PROFILING_EVENT_CL_GUID(name, guid)
const ConstTensorHandle * m_LayerOutput
void Execute() const override
Copyright (c) 2021 ARM Limited and Contributors.
ClConstantWorkload(const ConstantQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
#define ARMNN_ASSERT_MSG(COND, MSG)
arm_compute::Status ClConstantWorkloadValidate(const TensorInfo &output)
ConstantQueueDescriptor m_Data
#define ARMNN_ASSERT(COND)
profiling::ProfilingGuid GetGuid() const final
void CopyArmComputeClTensorData(arm_compute::CLTensor &dstTensor, const T *srcData)
std::vector< ITensorHandle * > m_Outputs
Contains information about TensorInfos of a layer.
const T * GetConstTensor() const