From fd627ffaec8fd8801d980b4c91ee7c0607ab6aaf Mon Sep 17 00:00:00 2001 From: Jan Eilers Date: Thu, 25 Feb 2021 17:44:00 +0000 Subject: IVGCVSW-5687 Update Doxygen Docu * Update Doxygen Documentation for 21.02 release Signed-off-by: Jan Eilers Change-Id: I9ed2f9caab038836ea99d7b378d7899fe431a4e5 --- 21.02/_ref_workload_utils_8hpp_source.xhtml | 143 ++++++++++++++++++++++++++++ 1 file changed, 143 insertions(+) create mode 100644 21.02/_ref_workload_utils_8hpp_source.xhtml (limited to '21.02/_ref_workload_utils_8hpp_source.xhtml') diff --git a/21.02/_ref_workload_utils_8hpp_source.xhtml b/21.02/_ref_workload_utils_8hpp_source.xhtml new file mode 100644 index 0000000000..8c0b367f45 --- /dev/null +++ b/21.02/_ref_workload_utils_8hpp_source.xhtml @@ -0,0 +1,143 @@ + + + + + + + + + + + + + +ArmNN: src/backends/reference/workloads/RefWorkloadUtils.hpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  21.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
RefWorkloadUtils.hpp
+
+
+Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include <backendsCommon/CpuTensorHandle.hpp>
9 
10 #include <armnn/Tensor.hpp>
11 #include <armnn/Types.hpp>
12 #include <armnn/utility/PolymorphicDowncast.hpp>
13 
14 #include <reference/RefTensorHandle.hpp>
15 
16 #include <BFloat16.hpp>
17 #include <Half.hpp>
18 
19 namespace armnn
20 {
21 
22 ////////////////////////////////////////////
23 /// float32 helpers
24 ////////////////////////////////////////////
25 
26 inline const TensorInfo& GetTensorInfo(const ITensorHandle* tensorHandle)
27 {
28  // We know that reference workloads use RefTensorHandles for inputs and outputs
29  const RefTensorHandle* refTensorHandle =
30  PolymorphicDowncast<const RefTensorHandle*>(tensorHandle);
31  return refTensorHandle->GetTensorInfo();
32 }
33 
34 template <typename DataType, typename PayloadType>
35 const DataType* GetInputTensorData(unsigned int idx, const PayloadType& data)
36 {
37  const ITensorHandle* tensorHandle = data.m_Inputs[idx];
38  return reinterpret_cast<const DataType*>(tensorHandle->Map());
39 }
40 
41 template <typename DataType, typename PayloadType>
42 DataType* GetOutputTensorData(unsigned int idx, const PayloadType& data)
43 {
44  ITensorHandle* tensorHandle = data.m_Outputs[idx];
45  return reinterpret_cast<DataType*>(tensorHandle->Map());
46 }
47 
48 template <typename PayloadType>
49 const float* GetInputTensorDataFloat(unsigned int idx, const PayloadType& data)
50 {
51  return GetInputTensorData<float>(idx, data);
52 }
53 
54 template <typename PayloadType>
55 float* GetOutputTensorDataFloat(unsigned int idx, const PayloadType& data)
56 {
57  return GetOutputTensorData<float>(idx, data);
58 }
59 
60 template <typename PayloadType>
61 const Half* GetInputTensorDataHalf(unsigned int idx, const PayloadType& data)
62 {
63  return GetInputTensorData<Half>(idx, data);
64 }
65 
66 template <typename PayloadType>
67 Half* GetOutputTensorDataHalf(unsigned int idx, const PayloadType& data)
68 {
69  return GetOutputTensorData<Half>(idx, data);
70 }
71 
72 template <typename PayloadType>
73 const BFloat16* GetInputTensorDataBFloat16(unsigned int idx, const PayloadType& data)
74 {
75  return GetInputTensorData<BFloat16>(idx, data);
76 }
77 
78 template <typename PayloadType>
79 BFloat16* GetOutputTensorDataBFloat16(unsigned int idx, const PayloadType& data)
80 {
81  return GetOutputTensorData<BFloat16>(idx, data);
82 }
83 
84 ////////////////////////////////////////////
85 /// u8 helpers
86 ////////////////////////////////////////////
87 
88 template<typename T>
89 std::vector<float> Dequantize(const T* quant, const TensorInfo& info)
90 {
91  std::vector<float> ret(info.GetNumElements());
92  for (size_t i = 0; i < info.GetNumElements(); i++)
93  {
94  ret[i] = armnn::Dequantize(quant[i], info.GetQuantizationScale(), info.GetQuantizationOffset());
95  }
96  return ret;
97 }
98 
99 template<typename T>
100 inline void Dequantize(const T* inputData, float* outputData, const TensorInfo& info)
101 {
102  for (unsigned int i = 0; i < info.GetNumElements(); i++)
103  {
104  outputData[i] = Dequantize<T>(inputData[i], info.GetQuantizationScale(), info.GetQuantizationOffset());
105  }
106 }
107 
108 inline void Quantize(uint8_t* quant, const float* dequant, const TensorInfo& info)
109 {
110  for (size_t i = 0; i < info.GetNumElements(); i++)
111  {
112  quant[i] = armnn::Quantize<uint8_t>(dequant[i], info.GetQuantizationScale(), info.GetQuantizationOffset());
113  }
114 }
115 
116 } //namespace armnn
armnn::GetInputTensorDataBFloat16
const BFloat16 * GetInputTensorDataBFloat16(unsigned int idx, const PayloadType &data)
Definition: RefWorkloadUtils.hpp:73
+
armnn::GetInputTensorDataFloat
const float * GetInputTensorDataFloat(unsigned int idx, const PayloadType &data)
Definition: RefWorkloadUtils.hpp:49
+
armnn::Dequantize
float Dequantize(QuantizedType value, float scale, int32_t offset)
Dequantize an 8-bit data type into a floating point data type.
Definition: TypesUtils.cpp:46
+
armnn::GetInputTensorData
const DataType * GetInputTensorData(unsigned int idx, const PayloadType &data)
Definition: RefWorkloadUtils.hpp:35
+
Tensor.hpp
+ + + +
armnn::GetOutputTensorDataHalf
Half * GetOutputTensorDataHalf(unsigned int idx, const PayloadType &data)
Definition: RefWorkloadUtils.hpp:67
+
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_software_tools.dox:6
+
armnn::GetInputTensorDataHalf
const Half * GetInputTensorDataHalf(unsigned int idx, const PayloadType &data)
Definition: RefWorkloadUtils.hpp:61
+
PolymorphicDowncast.hpp
+
armnn::DataType
DataType
Definition: Types.hpp:32
+
armnn::TensorInfo::GetQuantizationOffset
int32_t GetQuantizationOffset() const
Definition: Tensor.cpp:469
+ +
armnn::TensorInfo::GetQuantizationScale
float GetQuantizationScale() const
Definition: Tensor.cpp:452
+
Types.hpp
+
armnn::Quantize
QuantizedType Quantize(float value, float scale, int32_t offset)
Quantize a floating point data type into an 8-bit data type.
Definition: TypesUtils.cpp:30
+
RefTensorHandle.hpp
+
CpuTensorHandle.hpp
+
Half.hpp
+
armnn::ITensorHandle::Map
virtual const void * Map(bool blocking=true) const =0
Map the tensor data for access.
+
armnn::RefTensorHandle::GetTensorInfo
const TensorInfo & GetTensorInfo() const
Definition: RefTensorHandle.hpp:49
+
armnn::GetOutputTensorData
DataType * GetOutputTensorData(unsigned int idx, const PayloadType &data)
Definition: ClWorkloadUtils.hpp:147
+
BFloat16.hpp
+
armnn::GetOutputTensorDataFloat
float * GetOutputTensorDataFloat(unsigned int idx, const PayloadType &data)
Definition: RefWorkloadUtils.hpp:55
+
armnn::GetOutputTensorDataBFloat16
BFloat16 * GetOutputTensorDataBFloat16(unsigned int idx, const PayloadType &data)
Definition: RefWorkloadUtils.hpp:79
+
armnn::Half
half_float::half Half
Definition: Half.hpp:16
+
armnn::GetTensorInfo
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
Definition: RefWorkloadUtils.hpp:26
+
armnn::TensorInfo::GetNumElements
unsigned int GetNumElements() const
Definition: Tensor.hpp:192
+
+
+ + + + -- cgit v1.2.1