MockTensorHandleFactory Class Reference

#include <MockTensorHandleFactory.hpp>

Inheritance diagram for MockTensorHandleFactory:

Collaboration diagram for MockTensorHandleFactory:

Public Member Functions
	MockTensorHandleFactory (std::shared_ptr< MockMemoryManager > mgr)
std::unique_ptr< ITensorHandle >	CreateSubTensorHandle (ITensorHandle &parent, TensorShape const &subTensorShape, unsigned int const *subTensorOrigin) const override
std::unique_ptr< ITensorHandle >	CreateTensorHandle (const TensorInfo &tensorInfo) const override
std::unique_ptr< ITensorHandle >	CreateTensorHandle (const TensorInfo &tensorInfo, DataLayout dataLayout) const override
std::unique_ptr< ITensorHandle >	CreateTensorHandle (const TensorInfo &tensorInfo, const bool IsMemoryManaged) const override
	Utility Functions for backends which require TensorHandles to have unmanaged memory. More...
std::unique_ptr< ITensorHandle >	CreateTensorHandle (const TensorInfo &tensorInfo, DataLayout dataLayout, const bool IsMemoryManaged) const override
const FactoryId &	GetId () const override
bool	SupportsSubTensors () const override
MemorySourceFlags	GetExportFlags () const override
MemorySourceFlags	GetImportFlags () const override
Public Member Functions inherited from ITensorHandleFactory
virtual	~ITensorHandleFactory ()
	Some TensorHandleFactory decisions are deferred to run-time. More...
virtual bool	SupportsInPlaceComputation () const
virtual bool	SupportsMapUnmap () const
virtual std::vector< Capability >	GetCapabilities (const IConnectableLayer *layer, const IConnectableLayer *connectedLayer, CapabilityClass capabilityClass)

Static Public Member Functions
static const FactoryId &	GetIdStatic ()

Additional Inherited Members
Public Types inherited from ITensorHandleFactory
using	FactoryId = std::string
Static Public Attributes inherited from ITensorHandleFactory
static const FactoryId	LegacyFactoryId = "armnn_legacy_factory"
static const FactoryId	DeferredFactoryId = "armnn_deferred_factory"
	Use the workload factory to create the tensor handle. More...

Detailed Description

Definition at line 19 of file MockTensorHandleFactory.hpp.

Constructor & Destructor Documentation

MockTensorHandleFactory()

MockTensorHandleFactory ( std::shared_ptr< MockMemoryManager >  mgr )

inlineexplicit

Definition at line 23 of file MockTensorHandleFactory.hpp.

        : m_MemoryManager(mgr)
        , m_ImportFlags(static_cast<MemorySourceFlags>(MemorySource::Malloc))
        , m_ExportFlags(static_cast<MemorySourceFlags>(MemorySource::Malloc))
    {}

References armnn::Malloc.

Member Function Documentation

CreateSubTensorHandle()

std::unique_ptr< ITensorHandle > CreateSubTensorHandle ( ITensorHandle &  parent, TensorShape const &  subTensorShape, unsigned int const *  subTensorOrigin  ) const

overridevirtual

Implements ITensorHandleFactory.

Definition at line 20 of file MockTensorHandleFactory.cpp.

{
    return nullptr;
}

CreateTensorHandle() [1/4]

std::unique_ptr< ITensorHandle > CreateTensorHandle ( const TensorInfo &  tensorInfo ) const

overridevirtual

Implements ITensorHandleFactory.

Definition at line 27 of file MockTensorHandleFactory.cpp.

{
    return std::make_unique<MockTensorHandle>(tensorInfo, m_MemoryManager);
}

CreateTensorHandle() [2/4]

std::unique_ptr< ITensorHandle > CreateTensorHandle ( const TensorInfo &  tensorInfo, const bool  IsMemoryManaged  ) const

overridevirtual

Utility Functions for backends which require TensorHandles to have unmanaged memory.

These should be overloaded if required to facilitate direct import of input tensors and direct export of output tensors.

Reimplemented from ITensorHandleFactory.

Definition at line 39 of file MockTensorHandleFactory.cpp.

{
    if (IsMemoryManaged)
    {
        return std::make_unique<MockTensorHandle>(tensorInfo, m_MemoryManager);
    }
    else
    {
        return std::make_unique<MockTensorHandle>(tensorInfo, m_ImportFlags);
    }
}

CreateTensorHandle() [3/4]

std::unique_ptr< ITensorHandle > CreateTensorHandle ( const TensorInfo &  tensorInfo, DataLayout  dataLayout  ) const

overridevirtual

Implements ITensorHandleFactory.

Definition at line 32 of file MockTensorHandleFactory.cpp.

{
    IgnoreUnused(dataLayout);
    return std::make_unique<MockTensorHandle>(tensorInfo, m_MemoryManager);
}

References armnn::IgnoreUnused().

CreateTensorHandle() [4/4]

std::unique_ptr< ITensorHandle > CreateTensorHandle ( const TensorInfo &  tensorInfo, DataLayout  dataLayout, const bool  IsMemoryManaged  ) const

overridevirtual

Reimplemented from ITensorHandleFactory.

Definition at line 52 of file MockTensorHandleFactory.cpp.

{
    IgnoreUnused(dataLayout);
    if (IsMemoryManaged)
    {
        return std::make_unique<MockTensorHandle>(tensorInfo, m_MemoryManager);
    }
    else
    {
        return std::make_unique<MockTensorHandle>(tensorInfo, m_ImportFlags);
    }
}

References armnn::IgnoreUnused().

GetExportFlags()

MemorySourceFlags GetExportFlags ( ) const

overridevirtual

Reimplemented from ITensorHandleFactory.

Definition at line 77 of file MockTensorHandleFactory.cpp.

{
    return m_ExportFlags;
}

GetId()

const FactoryId & GetId ( ) const

overridevirtual

Implements ITensorHandleFactory.

Definition at line 67 of file MockTensorHandleFactory.cpp.

{
    return GetIdStatic();
}

References MockTensorHandleFactory::GetIdStatic().

GetIdStatic()

const FactoryId & GetIdStatic ( )

static

Definition at line 14 of file MockTensorHandleFactory.cpp.

{
    static const FactoryId s_Id(MockTensorHandleFactoryId());
    return s_Id;
}

References armnn::MockTensorHandleFactoryId().

Referenced by MockTensorHandleFactory::GetId().

GetImportFlags()

MemorySourceFlags GetImportFlags ( ) const

overridevirtual

Reimplemented from ITensorHandleFactory.

Definition at line 82 of file MockTensorHandleFactory.cpp.

{
    return m_ImportFlags;
}

SupportsSubTensors()

bool SupportsSubTensors ( ) const

overridevirtual

Implements ITensorHandleFactory.

Definition at line 72 of file MockTensorHandleFactory.cpp.

{
    return false;
}

---

The documentation for this class was generated from the following files:

• src/armnnTestUtils/MockTensorHandleFactory.hpp
• src/armnnTestUtils/MockTensorHandleFactory.cpp