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