MCPcopy Create free account
hub / github.com/ARM-software/armnn / GetTensor

Method GetTensor

include/armnn/backends/TensorHandle.hpp:83–93  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

81public:
82 template <typename T>
83 T* GetTensor() const
84 {
85 if (armnnUtils::CompatibleTypes<T>(GetTensorInfo().GetDataType()))
86 {
87 return reinterpret_cast<T*>(m_MutableMemory);
88 }
89 else
90 {
91 throw armnn::Exception("Attempting to get not compatible type tensor!");
92 }
93 }
94
95protected:
96 TensorHandle(const TensorInfo& tensorInfo);

Callers 15

CopyOutToMethod · 0.80
CopyInFromMethod · 0.80
CopyOutToMethod · 0.80
CopyInFromMethod · 0.80
CopyOutToMethod · 0.80
CopyInFromMethod · 0.80
ClSubTensorHandleMethod · 0.80
CopyOutToMethod · 0.80
CopyInFromMethod · 0.80
CopyOutToMethod · 0.80

Calls 3

GetTensorInfoFunction · 0.85
ExceptionClass · 0.50
GetDataTypeMethod · 0.45

Tested by 3

ClSplitterConcatTestFunction · 0.64
TestNeonTensorHandleInfoFunction · 0.64
CompareTensorHandleShapeFunction · 0.64