22.11
|
#include <TosaRefWorkloadFactory.hpp>
Public Member Functions | |
TosaRefWorkloadFactory (const std::shared_ptr< TosaRefMemoryManager > &memoryManager) | |
TosaRefWorkloadFactory () | |
~TosaRefWorkloadFactory () | |
const BackendId & | GetBackendId () const override |
bool | SupportsSubTensors () const override |
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 bool IsMemoryManaged=true) const override |
std::unique_ptr< ITensorHandle > | CreateTensorHandle (const TensorInfo &tensorInfo, DataLayout dataLayout, const bool IsMemoryManaged=true) const override |
std::unique_ptr< IWorkload > | CreateWorkload (LayerType type, const QueueDescriptor &descriptor, const WorkloadInfo &info) const override |
Public Member Functions inherited from IWorkloadFactory | |
virtual | ~IWorkloadFactory () |
virtual void | AfterWorkloadsCreated () |
Static Public Member Functions | |
static bool | IsLayerSupported (const Layer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported) |
static bool | IsLayerSupported (const IConnectableLayer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported, const ModelOptions &modelOptions) |
Static Public Member Functions inherited from IWorkloadFactory | |
static bool | IsLayerSupported (const BackendId &backendId, const IConnectableLayer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported) |
static bool | IsLayerSupported (const IConnectableLayer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported) |
static bool | IsLayerSupported (const IConnectableLayer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported, const ModelOptions &modelOptions) |
static bool | IsLayerSupported (const BackendId &backendId, const IConnectableLayer &layer, Optional< DataType > dataType, std::string &outReasonIfUnsupported, const ModelOptions &modelOptions) |
Definition at line 18 of file TosaRefWorkloadFactory.hpp.
|
explicit |
Definition at line 49 of file TosaRefWorkloadFactory.cpp.
Definition at line 54 of file TosaRefWorkloadFactory.cpp.
|
inline |
Definition at line 24 of file TosaRefWorkloadFactory.hpp.
References TosaRefWorkloadFactory::GetBackendId(), and TosaRefWorkloadFactory::IsLayerSupported().
|
inlineoverridevirtual |
Implements IWorkloadFactory.
Definition at line 40 of file TosaRefWorkloadFactory.hpp.
References ARMNN_DEPRECATED_MSG, TosaRefWorkloadFactory::CreateTensorHandle(), TosaRefWorkloadFactory::CreateWorkload(), armnn::IgnoreUnused(), and armnn::info.
|
overridevirtual |
Implements IWorkloadFactory.
Definition at line 79 of file TosaRefWorkloadFactory.cpp.
References armnn::Malloc.
Referenced by TosaRefWorkloadFactory::CreateSubTensorHandle().
|
overridevirtual |
Implements IWorkloadFactory.
Definition at line 92 of file TosaRefWorkloadFactory.cpp.
References armnn::IgnoreUnused(), and armnn::Malloc.
|
overridevirtual |
Reimplemented from IWorkloadFactory.
Definition at line 110 of file TosaRefWorkloadFactory.cpp.
References armnn::info, and armnn::PreCompiled.
Referenced by TosaRefWorkloadFactory::CreateSubTensorHandle().
|
overridevirtual |
Implements IWorkloadFactory.
Definition at line 59 of file TosaRefWorkloadFactory.cpp.
Referenced by TosaRefWorkloadFactory::~TosaRefWorkloadFactory().
|
static |
Definition at line 64 of file TosaRefWorkloadFactory.cpp.
References IWorkloadFactory::IsLayerSupported().
Referenced by TosaRefWorkloadFactory::~TosaRefWorkloadFactory().
|
static |
Definition at line 71 of file TosaRefWorkloadFactory.cpp.
References IWorkloadFactory::IsLayerSupported().
|
inlineoverridevirtual |
Implements IWorkloadFactory.
Definition at line 37 of file TosaRefWorkloadFactory.hpp.
References ARMNN_DEPRECATED_MSG.