| 81 | public: |
| 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 | |
| 95 | protected: |
| 96 | TensorHandle(const TensorInfo& tensorInfo); |