Lines Matching refs:arm_compute
19 arm_compute::DataType GetArmComputeDataType(armnn::DataType dataType, bool multiScales) in GetArmComputeDataType()
24 return arm_compute::DataType::BFLOAT16; in GetArmComputeDataType()
26 return arm_compute::DataType::U8; in GetArmComputeDataType()
28 return arm_compute::DataType::F16; in GetArmComputeDataType()
30 return arm_compute::DataType::F32; in GetArmComputeDataType()
32 return arm_compute::DataType::QASYMM8_SIGNED; in GetArmComputeDataType()
34 return arm_compute::DataType::QASYMM8; in GetArmComputeDataType()
36 return arm_compute::DataType::QSYMM16; in GetArmComputeDataType()
38 return arm_compute::DataType::S64; in GetArmComputeDataType()
41 … return multiScales ? arm_compute::DataType::QSYMM8_PER_CHANNEL : arm_compute::DataType::QSYMM8; in GetArmComputeDataType()
44 return arm_compute::DataType::S32; in GetArmComputeDataType()
47 return arm_compute::DataType::UNKNOWN; in GetArmComputeDataType()
51 armnn::DataType GetArmNNDataType(arm_compute::DataType dataType) in GetArmNNDataType()
55 case arm_compute::DataType::BFLOAT16: in GetArmNNDataType()
57 case arm_compute::DataType::U8: in GetArmNNDataType()
59 case arm_compute::DataType::F16: in GetArmNNDataType()
61 case arm_compute::DataType::F32: in GetArmNNDataType()
63 case arm_compute::DataType::QASYMM8_SIGNED: in GetArmNNDataType()
65 case arm_compute::DataType::QASYMM8: in GetArmNNDataType()
67 case arm_compute::DataType::QSYMM16: in GetArmNNDataType()
69 case arm_compute::DataType::S64: in GetArmNNDataType()
71 case arm_compute::DataType::QSYMM8_PER_CHANNEL: in GetArmNNDataType()
73 case arm_compute::DataType::QSYMM8: in GetArmNNDataType()
75 case arm_compute::DataType::S32: in GetArmNNDataType()
83 arm_compute::Coordinates BuildArmComputeReductionCoordinates(size_t inputDimensions, in BuildArmComputeReductionCoordinates()
87 arm_compute::Coordinates outAclCoords; in BuildArmComputeReductionCoordinates()
125 arm_compute::TensorShape BuildArmComputeTensorShape(const armnn::TensorShape& tensorShape) in BuildArmComputeTensorShape()
127 arm_compute::TensorShape shape; in BuildArmComputeTensorShape()
176 arm_compute::TensorShape BuildArmComputeTensorShape(const armnn::TensorShape& tensorShape, unsigned… in BuildArmComputeTensorShape()
178 arm_compute::TensorShape shape; in BuildArmComputeTensorShape()
200 arm_compute::TensorInfo BuildArmComputeTensorInfo(const armnn::TensorInfo& tensorInfo) in BuildArmComputeTensorInfo()
203 … const arm_compute::TensorShape aclTensorShape = BuildArmComputeTensorShape(tensorInfo.GetShape()); in BuildArmComputeTensorInfo()
204 …const arm_compute::DataType aclDataType = GetArmComputeDataType(tensorInfo.GetDataType(), mu… in BuildArmComputeTensorInfo()
206 const arm_compute::QuantizationInfo aclQuantizationInfo = multiScales ? in BuildArmComputeTensorInfo()
207 arm_compute::QuantizationInfo(tensorInfo.GetQuantizationScales()) : in BuildArmComputeTensorInfo()
208 …arm_compute::QuantizationInfo(tensorInfo.GetQuantizationScale(), tensorInfo.GetQuantizationOffset(… in BuildArmComputeTensorInfo()
210 return arm_compute::TensorInfo(aclTensorShape, 1, aclDataType, aclQuantizationInfo); in BuildArmComputeTensorInfo()
213 arm_compute::TensorInfo BuildArmComputeTensorInfo(const armnn::TensorInfo& tensorInfo, in BuildArmComputeTensorInfo()
216 arm_compute::TensorInfo aclTensorInfo = BuildArmComputeTensorInfo(tensorInfo); in BuildArmComputeTensorInfo()
222 arm_compute::TensorInfo BuildArmComputeTensorInfo(const armnn::TensorInfo& tensorInfo, unsigned int… in BuildArmComputeTensorInfo()
225 …const arm_compute::TensorShape aclTensorShape = BuildArmComputeTensorShape(tensorInfo.GetShape(), … in BuildArmComputeTensorInfo()
226 …const arm_compute::DataType aclDataType = GetArmComputeDataType(tensorInfo.GetDataType(), mu… in BuildArmComputeTensorInfo()
228 const arm_compute::QuantizationInfo aclQuantizationInfo = multiScales ? in BuildArmComputeTensorInfo()
229 arm_compute::QuantizationInfo(tensorInfo.GetQuantizationScales()) : in BuildArmComputeTensorInfo()
230 …arm_compute::QuantizationInfo(tensorInfo.GetQuantizationScale(), tensorInfo.GetQuantizationOffset(… in BuildArmComputeTensorInfo()
232 return arm_compute::TensorInfo(aclTensorShape, 1, aclDataType, aclQuantizationInfo); in BuildArmComputeTensorInfo()
234 arm_compute::TensorInfo BuildArmComputeTensorInfo(const armnn::TensorInfo& tensorInfo, in BuildArmComputeTensorInfo()
237 arm_compute::TensorInfo aclTensorInfo = BuildArmComputeTensorInfo(tensorInfo, dimensions); in BuildArmComputeTensorInfo()
244 arm_compute::DataLayout ConvertDataLayout(armnn::DataLayout dataLayout) in ConvertDataLayout()
248 case armnn::DataLayout::NHWC : return arm_compute::DataLayout::NHWC; in ConvertDataLayout()
250 case armnn::DataLayout::NCHW : return arm_compute::DataLayout::NCHW; in ConvertDataLayout()
252 case armnn::DataLayout::NDHWC : return arm_compute::DataLayout::NDHWC; in ConvertDataLayout()
254 case armnn::DataLayout::NCDHW : return arm_compute::DataLayout::NCDHW; in ConvertDataLayout()
261 arm_compute::PoolingLayerInfo BuildArmComputePoolingLayerInfo(const Pooling2dDescriptor& descriptor, in BuildArmComputePoolingLayerInfo()
265 …const arm_compute::PoolingType poolingType = ConvertPoolingAlgorithmToAclPoolingType(descriptor.m_… in BuildArmComputePoolingLayerInfo()
267 const arm_compute::DataLayout dataLayout = ConvertDataLayout(descriptor.m_DataLayout); in BuildArmComputePoolingLayerInfo()
273 return arm_compute::PoolingLayerInfo(poolingType, dataLayout); in BuildArmComputePoolingLayerInfo()
276 …const arm_compute::DimensionRoundingType rounding = ConvertOutputShapeRoundingToAclDimensionRoundi… in BuildArmComputePoolingLayerInfo()
278 const arm_compute::PadStrideInfo padStrideInfo(descriptor.m_StrideX, in BuildArmComputePoolingLayerInfo()
288 const arm_compute::Size2D poolSize(descriptor.m_PoolWidth, descriptor.m_PoolHeight); in BuildArmComputePoolingLayerInfo()
290 …return arm_compute::PoolingLayerInfo(poolingType, poolSize, dataLayout, padStrideInfo, excludePadd… in BuildArmComputePoolingLayerInfo()
294 arm_compute::Pooling3dLayerInfo BuildArmComputePooling3dLayerInfo(const Pooling3dDescriptor& descri… in BuildArmComputePooling3dLayerInfo()
297 …const arm_compute::PoolingType poolingType = ConvertPoolingAlgorithmToAclPoolingType(descriptor.m_… in BuildArmComputePooling3dLayerInfo()
303 return arm_compute::Pooling3dLayerInfo(poolingType); in BuildArmComputePooling3dLayerInfo()
306 …const arm_compute::Size3D poolSize(descriptor.m_PoolWidth, descriptor.m_PoolHeight, descriptor.m_P… in BuildArmComputePooling3dLayerInfo()
308 const arm_compute::Size3D stride(descriptor.m_StrideX, in BuildArmComputePooling3dLayerInfo()
312 const arm_compute::Padding3D padding(descriptor.m_PadLeft, in BuildArmComputePooling3dLayerInfo()
321 …const arm_compute::DimensionRoundingType rounding = ConvertOutputShapeRoundingToAclDimensionRoundi… in BuildArmComputePooling3dLayerInfo()
324 return arm_compute::Pooling3dLayerInfo(poolingType, in BuildArmComputePooling3dLayerInfo()
333 arm_compute::NormalizationLayerInfo BuildArmComputeNormalizationLayerInfo(const NormalizationDescri… in BuildArmComputeNormalizationLayerInfo()
335 const arm_compute::NormType normType = in BuildArmComputeNormalizationLayerInfo()
337 return arm_compute::NormalizationLayerInfo(normType, in BuildArmComputeNormalizationLayerInfo()
345 arm_compute::PermutationVector BuildArmComputePermutationVector(const armnn::PermutationVector& per… in BuildArmComputePermutationVector()
347 arm_compute::PermutationVector aclPerm; in BuildArmComputePermutationVector()
362 arm_compute::PermutationVector BuildArmComputeTransposeVector(const armnn::PermutationVector& perm) in BuildArmComputeTransposeVector()
386 arm_compute::PermutationVector aclPerm; in BuildArmComputeTransposeVector()
406 arm_compute::Size2D BuildArmComputeSize2D(const unsigned int width, const unsigned int height) in BuildArmComputeSize2D()
408 return arm_compute::Size2D(width, height); in BuildArmComputeSize2D()
411 arm_compute::PixelValue GetPixelValue(const arm_compute::ITensorInfo* tensorInfo, float value) in GetPixelValue()
415 case arm_compute::DataType::F16: in GetPixelValue()
417 arm_compute::PixelValue pixelValue = arm_compute::PixelValue(static_cast<Half>(value)); in GetPixelValue()
424 case arm_compute::DataType::F32: in GetPixelValue()
425 return arm_compute::PixelValue(value); in GetPixelValue()
426 case arm_compute::DataType::QASYMM8: in GetPixelValue()
427 return arm_compute::PixelValue(static_cast<uint8_t>(value)); in GetPixelValue()
428 case arm_compute::DataType::QSYMM16: in GetPixelValue()
429 return arm_compute::PixelValue(static_cast<int16_t>(value)); in GetPixelValue()
430 case arm_compute::DataType::QSYMM8: in GetPixelValue()
431 case arm_compute::DataType::QASYMM8_SIGNED: in GetPixelValue()
432 case arm_compute::DataType::QSYMM8_PER_CHANNEL: in GetPixelValue()
433 return arm_compute::PixelValue(static_cast<int8_t>(value)); in GetPixelValue()
434 case arm_compute::DataType::S32: in GetPixelValue()
435 return arm_compute::PixelValue(static_cast<int32_t>(value)); in GetPixelValue()
443 const arm_compute::TensorShape& weightsShape, in ComputeDepthwiseConv2dDepthMultiplier()
444 const arm_compute::TensorShape& inputShape) in ComputeDepthwiseConv2dDepthMultiplier()