diff options
Diffstat (limited to 'src/backends/ClTensorHandle.hpp')
-rw-r--r-- | src/backends/ClTensorHandle.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/backends/ClTensorHandle.hpp b/src/backends/ClTensorHandle.hpp index 9c78192284..e3d7b5b491 100644 --- a/src/backends/ClTensorHandle.hpp +++ b/src/backends/ClTensorHandle.hpp @@ -37,6 +37,11 @@ public: armnn::armcomputetensorutils::BuildArmComputeTensor(m_Tensor, tensorInfo); } + ClTensorHandle(const TensorInfo& tensorInfo, DataLayout dataLayout) + { + armnn::armcomputetensorutils::BuildArmComputeTensor(m_Tensor, tensorInfo, dataLayout); + } + arm_compute::CLTensor& GetTensor() override { return m_Tensor; } arm_compute::CLTensor const& GetTensor() const override { return m_Tensor; } virtual void Allocate() override {armnn::armcomputetensorutils::InitialiseArmComputeTensorEmpty(m_Tensor);} |