#include <IRuntime.hpp>
|
virtual Status | LoadNetwork (NetworkId &networkIdOut, IOptimizedNetworkPtr network)=0 |
| Loads a complete network into the IRuntime. More...
|
|
virtual Status | LoadNetwork (NetworkId &networkIdOut, IOptimizedNetworkPtr network, std::string &errorMessage)=0 |
| Load a complete network into the IRuntime. More...
|
|
virtual Status | LoadNetwork (NetworkId &networkIdOut, IOptimizedNetworkPtr network, std::string &errorMessage, const INetworkProperties &networkProperties)=0 |
|
virtual TensorInfo | GetInputTensorInfo (NetworkId networkId, LayerBindingId layerId) const =0 |
|
virtual TensorInfo | GetOutputTensorInfo (NetworkId networkId, LayerBindingId layerId) const =0 |
|
virtual Status | EnqueueWorkload (NetworkId networkId, const InputTensors &inputTensors, const OutputTensors &outputTensors)=0 |
| Evaluates a network using input in inputTensors and outputs filled into outputTensors. More...
|
|
virtual Status | UnloadNetwork (NetworkId networkId)=0 |
| Unloads a network from the IRuntime. More...
|
|
virtual const IDeviceSpec & | GetDeviceSpec () const =0 |
|
virtual const std::shared_ptr< IProfiler > | GetProfiler (NetworkId networkId) const =0 |
| Gets the profiler corresponding to the given network id. More...
|
|
virtual void | RegisterDebugCallback (NetworkId networkId, const DebugCallbackFunction &func)=0 |
| Registers a callback function to debug layers performing custom computations on intermediate tensors. More...
|
|
Definition at line 38 of file IRuntime.hpp.
◆ ~IRuntime()
◆ Create()
◆ CreateRaw()
Definition at line 27 of file Runtime.cpp.
armnn::Runtime::CreationOptions::ExternalProfilingOptions options
◆ Destroy()
◆ EnqueueWorkload()
Evaluates a network using input in inputTensors and outputs filled into outputTensors.
Implemented in Runtime.
◆ GetDeviceSpec()
◆ GetInputTensorInfo()
◆ GetOutputTensorInfo()
◆ GetProfiler()
Gets the profiler corresponding to the given network id.
- Parameters
-
networkId | The id of the network for which to get the profile. |
- Returns
- A pointer to the requested profiler, or nullptr if not found.
Implemented in Runtime.
◆ LoadNetwork() [1/3]
Loads a complete network into the IRuntime.
- Parameters
-
[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. |
- Returns
- armnn::Status
Implemented in Runtime.
◆ LoadNetwork() [2/3]
Load a complete network into the IRuntime.
- Parameters
-
[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. |
- Returns
- armnn::Status
Implemented in Runtime.
◆ LoadNetwork() [3/3]
◆ RegisterDebugCallback()
Registers a callback function to debug layers performing custom computations on intermediate tensors.
- Parameters
-
networkId | The id of the network to register the callback. |
func | callback function to pass to the debug layer. |
Implemented in Runtime.
◆ UnloadNetwork()
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.
- Parameters
-
[in] | networkId | - Unique identifier for the network to be unloaded. Generated in LoadNetwork(). |
- Returns
- armnn::Status
Implemented in Runtime.
The documentation for this class was generated from the following files: