diff options
Diffstat (limited to 'src/armnn/Runtime.cpp')
-rw-r--r-- | src/armnn/Runtime.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/armnn/Runtime.cpp b/src/armnn/Runtime.cpp index 09be92c709..f8b2462f96 100644 --- a/src/armnn/Runtime.cpp +++ b/src/armnn/Runtime.cpp @@ -231,4 +231,10 @@ Status Runtime::EnqueueWorkload(NetworkId networkId, return loadedNetwork->EnqueueWorkload(inputTensors, outputTensors); } +void Runtime::RegisterDebugCallback(NetworkId networkId, const DebugCallbackFunction& func) +{ + LoadedNetwork* loadedNetwork = GetLoadedNetworkPtr(networkId); + loadedNetwork->RegisterDebugCallback(func); +} + } |