ArmNN
NotReleased
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
8
#include <
armnn/backends/ITensorHandle.hpp
>
9
10
#include <arm_compute/runtime/IMemoryGroup.h>
11
#include <arm_compute/runtime/Tensor.h>
12
13
namespace
armnn
14
{
15
16
class
IAclTensorHandle
:
public
ITensorHandle
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::SetMemoryGroup
virtual void SetMemoryGroup(const std::shared_ptr< arm_compute::IMemoryGroup > &memoryGroup)=0
armnn::IAclTensorHandle::GetDataType
virtual arm_compute::DataType GetDataType() const =0
ITensorHandle.hpp
armnn::ITensorHandle
Definition:
ITensorHandle.hpp:16
armnn::IAclTensorHandle
Definition:
ArmComputeTensorHandle.hpp:16
armnn
Definition:
BackendHelper.hpp:11
armnn::IAclTensorHandle::GetTensor
virtual arm_compute::ITensor & GetTensor()=0
armnn::DataType
DataType
Definition:
Types.hpp:32
src
backends
aclCommon
ArmComputeTensorHandle.hpp
Generated on Fri Mar 13 2020 16:06:56 for ArmNN by
1.8.13