30 if (armnnUtils::CompatibleTypes<T>(
GetTensorInfo().GetDataType()))
32 return reinterpret_cast<const T*
>(m_Memory);
49 virtual const void*
Map(
bool )
const override {
return m_Memory; }
50 virtual void Unmap()
const override {}
65 void CopyOutTo(
void *)
const override {
ARMNN_ASSERT_MSG(
false,
"Unimplemented"); }
66 void CopyInFrom(
const void*)
override {
ARMNN_ASSERT_MSG(
false,
"Unimplemented"); }
76 const void* ConstTensorHandle::GetConstTensor<void>()
const;
85 if (armnnUtils::CompatibleTypes<T>(
GetTensorInfo().GetDataType()))
87 return reinterpret_cast<T*
>(m_MutableMemory);
100 m_MutableMemory = mem;
108 void* m_MutableMemory;
112 void* TensorHandle::GetTensor<void>()
const;
134 void CopyOutTo(
void* memory)
const override;
135 void CopyInFrom(
const void* memory)
override;
138 void CopyFrom(
const void* srcMemory,
unsigned int numBytes);
193 , m_TensorHandle(
std::move(ptr)) {};
196 const void*
Map(
bool blocking =
true)
200 auto pRet = m_TensorHandle->Map(blocking);
232 if (m_Mapped && m_TensorHandle)
234 m_TensorHandle->Unmap();
241 return m_TensorHandle->GetTensorInfo();
251 std::shared_ptr<ConstTensorHandle> m_TensorHandle;
virtual const void * Map(bool) const override
Map the tensor data for access.
const TensorShape & GetShape() const
ConstPassthroughTensorHandle(const TensorInfo &tensorInfo, const void *mem)
void SetMemory(void *mem)
virtual void Allocate()=0
Indicate to the memory manager that this resource is no longer active.
void SetConstMemory(const void *mem)
TensorShape GetShape() const override
Get the number of elements for each dimension ordered from slowest iterating dimension to fastest ite...
const TensorInfo & GetTensorInfo() const
Copyright (c) 2021 ARM Limited and Contributors.
const TensorInfo & GetTensorInfo() const
ConstTensorHandle(const TensorInfo &tensorInfo)
#define ARMNN_ASSERT_MSG(COND, MSG)
A tensor defined by a TensorInfo (shape and data type) and an immutable backing store.
ManagedConstTensorHandle(std::shared_ptr< ConstTensorHandle > ptr)
TensorShape GetUnpaddedTensorStrides(const TensorInfo &tensorInfo)
virtual void Manage() override
Indicate to the memory manager that this resource is active.
~ManagedConstTensorHandle()
virtual void Unmap() const override
Unmap the tensor data.
Base class for all ArmNN exceptions so that users can filter to just those.
TensorShape GetStrides() const override
Get the strides for each dimension ordered from largest to smallest where the smallest value is the s...
PassthroughTensorHandle(const TensorInfo &tensorInfo, void *mem)
virtual ITensorHandle * GetParent() const override
Get the parent tensor if this is a subtensor.
const void * Map(bool blocking=true)
RAII Managed resource Unmaps MemoryArea once out of scope.
const T * GetConstTensor() const