ClTensorHandleFactory Class Reference

#include <ClTensorHandleFactory.hpp>

Inheritance diagram for ClTensorHandleFactory:

Public Member Functions
	ClTensorHandleFactory (std::shared_ptr< ClMemoryManager > mgr)
std::unique_ptr< ITensorHandle >	CreateSubTensorHandle (ITensorHandle &parent, const TensorShape &subTensorShape, const unsigned int *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=true) const override
std::unique_ptr< ITensorHandle >	CreateTensorHandle (const TensorInfo &tensorInfo, DataLayout dataLayout, const bool IsMemoryManaged=true) 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 ()
virtual bool	SupportsMapUnmap () const final

Static Public Member Functions
static const FactoryId &	GetIdStatic ()

Static Public Attributes
static const FactoryId	m_Id
Static Public Attributes inherited from ITensorHandleFactory
static const FactoryId	LegacyFactoryId = "armnn_legacy_factory"
static const FactoryId	DeferredFactoryId = "armnn_deferred_factory"

Additional Inherited Members
Public Types inherited from ITensorHandleFactory
using	FactoryId = std::string

Detailed Description

Definition at line 17 of file ClTensorHandleFactory.hpp.

Constructor & Destructor Documentation

ClTensorHandleFactory()

ClTensorHandleFactory ( std::shared_ptr< ClMemoryManager >  mgr )

inline

Definition at line 21 of file ClTensorHandleFactory.hpp.

References ClTensorHandleFactory::CreateSubTensorHandle(), ClTensorHandleFactory::CreateTensorHandle(), ClTensorHandleFactory::GetExportFlags(), ClTensorHandleFactory::GetId(), ClTensorHandleFactory::GetIdStatic(), ClTensorHandleFactory::GetImportFlags(), and ClTensorHandleFactory::SupportsSubTensors().

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

Member Function Documentation

CreateSubTensorHandle()

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

overridevirtual

Implements ITensorHandleFactory.

Definition at line 22 of file ClTensorHandleFactory.cpp.

Referenced by ClTensorHandleFactory::ClTensorHandleFactory().

{
    arm_compute::Coordinates coords;
    arm_compute::TensorShape shape = armcomputetensorutils::BuildArmComputeTensorShape(subTensorShape);

    coords.set_num_dimensions(subTensorShape.GetNumDimensions());
    for (unsigned int i = 0; i < subTensorShape.GetNumDimensions(); ++i)
    {
        // Arm compute indexes tensor coords in reverse order.
        unsigned int revertedIndex = subTensorShape.GetNumDimensions() - i - 1;
        coords.set(i, boost::numeric_cast<int>(subTensorOrigin[revertedIndex]));
    }

    const arm_compute::TensorShape parentShape = armcomputetensorutils::BuildArmComputeTensorShape(
            parent.GetShape());
    if (!::arm_compute::error_on_invalid_subtensor(__func__, __FILE__, __LINE__, parentShape, coords, shape))
    {
        return nullptr;
    }

    return std::make_unique<ClSubTensorHandle>(
            boost::polymorphic_downcast<IClTensorHandle *>(&parent), shape, coords);
}

CreateTensorHandle() [1/4]

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

overridevirtual

Implements ITensorHandleFactory.

Definition at line 48 of file ClTensorHandleFactory.cpp.

Referenced by ClTensorHandleFactory::ClTensorHandleFactory(), and ClTensorHandleFactory::CreateTensorHandle().

{
    return ClTensorHandleFactory::CreateTensorHandle(tensorInfo, true);
}

CreateTensorHandle() [2/4]

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

overridevirtual

Implements ITensorHandleFactory.

Definition at line 53 of file ClTensorHandleFactory.cpp.

References ClTensorHandleFactory::CreateTensorHandle().

{
    return ClTensorHandleFactory::CreateTensorHandle(tensorInfo, dataLayout, true);
}

CreateTensorHandle() [3/4]

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

overridevirtual

Reimplemented from ITensorHandleFactory.

Definition at line 59 of file ClTensorHandleFactory.cpp.

{
    std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo);
    if (IsMemoryManaged)
    {
        tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup());
    }
    return tensorHandle;
}

CreateTensorHandle() [4/4]

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

overridevirtual

Reimplemented from ITensorHandleFactory.

Definition at line 70 of file ClTensorHandleFactory.cpp.

{
    std::unique_ptr<ClTensorHandle> tensorHandle = std::make_unique<ClTensorHandle>(tensorInfo, dataLayout);
    if (IsMemoryManaged)
    {
        tensorHandle->SetMemoryGroup(m_MemoryManager->GetInterLayerMemoryGroup());
    }
    return tensorHandle;
}

GetExportFlags()

MemorySourceFlags GetExportFlags ( ) const

overridevirtual

Reimplemented from ITensorHandleFactory.

Definition at line 98 of file ClTensorHandleFactory.cpp.

Referenced by ClTensorHandleFactory::ClTensorHandleFactory().

{
    return m_ExportFlags;
}

GetId()

const FactoryId & GetId ( ) const

overridevirtual

Implements ITensorHandleFactory.

Definition at line 88 of file ClTensorHandleFactory.cpp.

References ClTensorHandleFactory::GetIdStatic().

Referenced by ClTensorHandleFactory::ClTensorHandleFactory().

{
    return GetIdStatic();
}

GetIdStatic()

const FactoryId & GetIdStatic ( )

static

Definition at line 82 of file ClTensorHandleFactory.cpp.

References armnn::ClTensorHandleFactoryId().

Referenced by ClTensorHandleFactory::ClTensorHandleFactory(), ClBackend::GetHandleFactoryPreferences(), and ClTensorHandleFactory::GetId().

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

GetImportFlags()

MemorySourceFlags GetImportFlags ( ) const

overridevirtual

Reimplemented from ITensorHandleFactory.

Definition at line 103 of file ClTensorHandleFactory.cpp.

Referenced by ClTensorHandleFactory::ClTensorHandleFactory().

{
    return m_ImportFlags;
}

SupportsSubTensors()

bool SupportsSubTensors ( ) const

overridevirtual

Implements ITensorHandleFactory.

Definition at line 93 of file ClTensorHandleFactory.cpp.

Referenced by ClTensorHandleFactory::ClTensorHandleFactory().

{
    return true;
}

Member Data Documentation

m_Id

const FactoryId m_Id

static

Definition at line 19 of file ClTensorHandleFactory.hpp.

---

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

• src/backends/cl/ClTensorHandleFactory.hpp
• src/backends/cl/ClTensorHandleFactory.cpp