ArmNN
 24.02
IClTensorHandle.hpp
Go to the documentation of this file.
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 
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
armnn::IClTensorHandle::SetMemoryGroup
virtual void SetMemoryGroup(const std::shared_ptr< arm_compute::IMemoryGroup > &memoryGroup)=0
armnn::IClTensorHandle::GetTensor
virtual arm_compute::ICLTensor & GetTensor()=0
armnn::IAclTensorHandle
Definition: ArmComputeTensorHandle.hpp:16
armnn::DataType
DataType
Definition: Types.hpp:48
armnn::IClTensorHandle::GetDataType
virtual arm_compute::DataType GetDataType() const =0
armnn::IClTensorHandle
Definition: IClTensorHandle.hpp:13
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6