diff options
author | surmeh01 <surabhi.mehta@arm.com> | 2018-03-29 16:33:54 +0100 |
---|---|---|
committer | surmeh01 <surabhi.mehta@arm.com> | 2018-03-29 16:33:54 +0100 |
commit | 7666005c72227a3ea5c410ca2861c9b6620887d8 (patch) | |
tree | 084296e0ba923f7885b8efb242335a4547b2cdb0 /ArmnnDriver.cpp | |
parent | 5307bc10ac488261e84ac76b2dede6039ea3fe96 (diff) | |
download | android-nn-driver-7666005c72227a3ea5c410ca2861c9b6620887d8.tar.gz |
Release 18.03
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 |