diff options
Diffstat (limited to 'ArmnnDriver.cpp')
-rw-r--r-- | ArmnnDriver.cpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/ArmnnDriver.cpp b/ArmnnDriver.cpp index 914d6560..19624649 100644 --- a/ArmnnDriver.cpp +++ b/ArmnnDriver.cpp @@ -372,6 +372,19 @@ Return<ErrorStatus> ArmnnDriver::prepareModel(const Model& model, return FailPrepareModel(ErrorStatus::GENERAL_FAILURE, message.str(), cb); } + // Check that the optimized network is valid. + if (!optNet) + { + return FailPrepareModel(ErrorStatus::GENERAL_FAILURE, + "ArmnnDriver::prepareModel: Invalid optimized network", cb); + } + + // Export the optimized network graph to a dot file if an output dump directory + // has been specified in the drivers' arguments. + ExportNetworkGraphToDotFile(*optNet, + m_Options.GetRequestInputsAndOutputsDumpDir(), + model); + // load it into the runtime armnn::NetworkId netId = 0; try |