|
25.11
|
#include <IRuntime.hpp>
Classes | |
| struct | CreationOptions |
Public Member Functions | |
| Status | LoadNetwork (NetworkId &networkIdOut, IOptimizedNetworkPtr network) |
| Loads a complete network into the IRuntime. | |
| Status | LoadNetwork (NetworkId &networkIdOut, IOptimizedNetworkPtr network, std::string &errorMessage) |
| Load a complete network into the IRuntime. | |
| Status | LoadNetwork (NetworkId &networkIdOut, IOptimizedNetworkPtr network, std::string &errorMessage, const INetworkProperties &networkProperties) |
| TensorInfo | GetInputTensorInfo (NetworkId networkId, LayerBindingId layerId) const |
| TensorInfo | GetOutputTensorInfo (NetworkId networkId, LayerBindingId layerId) const |
| std::vector< ImportedInputId > | ImportInputs (NetworkId networkId, const InputTensors &inputTensors, MemorySource forceImportMemorySource=MemorySource::Undefined) |
| ImportInputs separates the importing and mapping of InputTensors from network execution. | |
| std::vector< ImportedOutputId > | ImportOutputs (NetworkId networkId, const OutputTensors &outputTensors, MemorySource forceImportMemorySource=MemorySource::Undefined) |
| ImportOutputs separates the importing and mapping of OutputTensors from network execution. | |
| Status | EnqueueWorkload (NetworkId networkId, const InputTensors &inputTensors, const OutputTensors &outputTensors, std::vector< ImportedInputId > preImportedInputIds={}, std::vector< ImportedOutputId > preImportedOutputIds={}) |
| Evaluates a network using input in inputTensors and outputs filled into outputTensors. | |
| Status | UnloadNetwork (NetworkId networkId) |
| Unloads a network from the IRuntime. | |
| const IDeviceSpec & | GetDeviceSpec () const |
| const std::shared_ptr< IProfiler > | GetProfiler (NetworkId networkId) const |
| Gets the profiler corresponding to the given network id. | |
| void | RegisterDebugCallback (NetworkId networkId, const DebugCallbackFunction &func) |
| Registers a callback function to debug layers performing custom computations on intermediate tensors. | |
Static Public Member Functions | |
| static IRuntime * | CreateRaw (const CreationOptions &options) |
| static IRuntimePtr | Create (const CreationOptions &options) |
| static void | Destroy (IRuntime *runtime) |
Protected Member Functions | |
| IRuntime () | |
| IRuntime (const IRuntime::CreationOptions &options) | |
| ~IRuntime () | |
Protected Attributes | |
| std::unique_ptr< RuntimeImpl > | pRuntimeImpl |
Definition at line 67 of file IRuntime.hpp.
|
protected |
Definition at line 41 of file Runtime.cpp.
References IRuntime(), and pRuntimeImpl.
Referenced by CreateRaw(), Destroy(), and IRuntime().
|
protected |
|
protecteddefault |
|
static |
Definition at line 52 of file Runtime.cpp.
References CreateRaw(), and Destroy().
|
static |
|
static |
Definition at line 57 of file Runtime.cpp.
References IRuntime().
Referenced by Create().
| Status EnqueueWorkload | ( | NetworkId | networkId, |
| const InputTensors & | inputTensors, | ||
| const OutputTensors & | outputTensors, | ||
| std::vector< ImportedInputId > | preImportedInputIds = {}, | ||
| std::vector< ImportedOutputId > | preImportedOutputIds = {} ) |
Evaluates a network using input in inputTensors and outputs filled into outputTensors.
Definition at line 104 of file Runtime.cpp.
References pRuntimeImpl.
| const IDeviceSpec & GetDeviceSpec | ( | ) | const |
Definition at line 119 of file Runtime.cpp.
References pRuntimeImpl.
| armnn::TensorInfo GetInputTensorInfo | ( | NetworkId | networkId, |
| LayerBindingId | layerId ) const |
Definition at line 82 of file Runtime.cpp.
References pRuntimeImpl.
| armnn::TensorInfo GetOutputTensorInfo | ( | NetworkId | networkId, |
| LayerBindingId | layerId ) const |
Definition at line 87 of file Runtime.cpp.
References pRuntimeImpl.
Gets the profiler corresponding to the given network id.
| networkId | The id of the network for which to get the profile. |
Definition at line 124 of file Runtime.cpp.
References pRuntimeImpl.
| std::vector< ImportedInputId > ImportInputs | ( | NetworkId | networkId, |
| const InputTensors & | inputTensors, | ||
| MemorySource | forceImportMemorySource = MemorySource::Undefined ) |
ImportInputs separates the importing and mapping of InputTensors from network execution.
Allowing for a set of InputTensors to be imported and mapped once, but used in execution many times. This function is not thread safe and must not be used while other threads are calling Execute(). No exceptions are thrown for failed imports. It is the caller's responsibility to check whether tensors have been successfully imported by comparing returned ids with those passed in the InputTensors. Whether a tensor can be imported or not is backend specific.
Definition at line 92 of file Runtime.cpp.
References pRuntimeImpl.
| std::vector< ImportedOutputId > ImportOutputs | ( | NetworkId | networkId, |
| const OutputTensors & | outputTensors, | ||
| MemorySource | forceImportMemorySource = MemorySource::Undefined ) |
ImportOutputs separates the importing and mapping of OutputTensors from network execution.
Allowing for a set of OutputTensors to be imported and mapped once, but used in execution many times. This function is not thread safe and must not be used while other threads are calling Execute(). No exceptions are thrown for failed imports. It is the caller's responsibility to check whether tensors have been successfully imported by comparing returned ids with those passed in the OutputTensors. Whether a tensor can be imported or not is backend specific.
Definition at line 98 of file Runtime.cpp.
References pRuntimeImpl.
| Status LoadNetwork | ( | NetworkId & | networkIdOut, |
| IOptimizedNetworkPtr | network ) |
Loads a complete network into the IRuntime.
| [out] | networkIdOut | - Unique identifier for the network is returned in this reference. |
| [in] | network | - Complete network to load into the IRuntime. The runtime takes ownership of the network once passed in. |
Definition at line 62 of file Runtime.cpp.
References pRuntimeImpl.
| Status LoadNetwork | ( | NetworkId & | networkIdOut, |
| IOptimizedNetworkPtr | network, | ||
| std::string & | errorMessage ) |
Load a complete network into the IRuntime.
| [out] | networkIdOut | Unique identifier for the network is returned in this reference. |
| [in] | network | Complete network to load into the IRuntime. |
| [out] | errorMessage | Error message if there were any errors. The runtime takes ownership of the network once passed in. |
Definition at line 67 of file Runtime.cpp.
References pRuntimeImpl.
| Status LoadNetwork | ( | NetworkId & | networkIdOut, |
| IOptimizedNetworkPtr | network, | ||
| std::string & | errorMessage, | ||
| const INetworkProperties & | networkProperties ) |
Definition at line 74 of file Runtime.cpp.
References pRuntimeImpl.
| void RegisterDebugCallback | ( | NetworkId | networkId, |
| const DebugCallbackFunction & | func ) |
Registers a callback function to debug layers performing custom computations on intermediate tensors.
| networkId | The id of the network to register the callback. |
| func | callback function to pass to the debug layer. |
Definition at line 129 of file Runtime.cpp.
References pRuntimeImpl.
Unloads a network from the IRuntime.
At the moment this only removes the network from the m_Impl->m_Network. This might need more work in the future to be AndroidNN compliant.
| [in] | networkId | - Unique identifier for the network to be unloaded. Generated in LoadNetwork(). |
Definition at line 114 of file Runtime.cpp.
References pRuntimeImpl.
|
protected |
Definition at line 263 of file IRuntime.hpp.
Referenced by EnqueueWorkload(), GetDeviceSpec(), GetInputTensorInfo(), GetOutputTensorInfo(), GetProfiler(), ImportInputs(), ImportOutputs(), IRuntime(), IRuntime(), LoadNetwork(), LoadNetwork(), LoadNetwork(), RegisterDebugCallback(), and UnloadNetwork().