ArmNN
 24.02
ArmComputeTensorHandle.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
9 
10 #include <arm_compute/runtime/IMemoryGroup.h>
11 #include <arm_compute/runtime/Tensor.h>
12 
13 namespace armnn
14 {
15 
17 {
18 public:
19  virtual arm_compute::ITensor& GetTensor() = 0;
20  virtual arm_compute::ITensor const& GetTensor() const = 0;
21  virtual arm_compute::DataType GetDataType() const = 0;
22  virtual void SetMemoryGroup(const std::shared_ptr<arm_compute::IMemoryGroup>& memoryGroup) = 0;
23 };
24 
25 } //namespace armnn
armnn::IAclTensorHandle::GetDataType
virtual arm_compute::DataType GetDataType() const =0
armnn::IAclTensorHandle::SetMemoryGroup
virtual void SetMemoryGroup(const std::shared_ptr< arm_compute::IMemoryGroup > &memoryGroup)=0
armnn::ITensorHandle
Definition: ITensorHandle.hpp:16
ITensorHandle.hpp
armnn::IAclTensorHandle
Definition: ArmComputeTensorHandle.hpp:16
armnn::DataType
DataType
Definition: Types.hpp:48
armnn::IAclTensorHandle::GetTensor
virtual arm_compute::ITensor & GetTensor()=0
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6