1 // 2 // Copyright © 2022 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 #pragma once 6 7 #include <arm_compute/core/CL/ICLTensor.h> 8 #include <arm_compute/runtime/MemoryGroup.h> 9 10 namespace armnn 11 { 12 13 class IClTensorHandle : public IAclTensorHandle 14 { 15 public: 16 virtual arm_compute::ICLTensor& GetTensor() = 0; 17 virtual arm_compute::ICLTensor const& GetTensor() const = 0; 18 virtual arm_compute::DataType GetDataType() const = 0; 19 virtual void SetMemoryGroup(const std::shared_ptr<arm_compute::IMemoryGroup>& memoryGroup) = 0; 20 }; 21 22 } //namespace armnn