diff options
Diffstat (limited to 'ArmnnDriverImpl.cpp')
-rw-r--r-- | ArmnnDriverImpl.cpp | 16 |
1 files changed, 11 insertions, 5 deletions
diff --git a/ArmnnDriverImpl.cpp b/ArmnnDriverImpl.cpp index ef37cae6..917370cf 100644 --- a/ArmnnDriverImpl.cpp +++ b/ArmnnDriverImpl.cpp @@ -100,6 +100,11 @@ Return<V1_0::ErrorStatus> ArmnnDriverImpl<HalPolicy>::prepareModel( return V1_0::ErrorStatus::NONE; } + // Serialize the network graph to a .armnn file if an output directory + // has been specified in the drivers' arguments. + auto serializedNetworkFileName = + SerializeNetwork(*modelConverter.GetINetwork(), options.GetRequestInputsAndOutputsDumpDir()); + // Optimize the network armnn::IOptimizedNetworkPtr optNet(nullptr, nullptr); armnn::OptimizerOptions OptOptions; @@ -169,11 +174,12 @@ Return<V1_0::ErrorStatus> ArmnnDriverImpl<HalPolicy>::prepareModel( return V1_0::ErrorStatus::NONE; } - // Now that we have a networkId for the graph rename the dump file to use it - // so that we can associate the graph file and the input/output tensor dump files - RenameGraphDotFile(dotGraphFileName, - options.GetRequestInputsAndOutputsDumpDir(), - netId); + // Now that we have a networkId for the graph rename the exported files to use it + // so that we can associate the graph file and the input/output tensor exported files + RenameExportedFiles(serializedNetworkFileName, + dotGraphFileName, + options.GetRequestInputsAndOutputsDumpDir(), + netId); sp<ArmnnPreparedModel<HalPolicy>> preparedModel( new ArmnnPreparedModel<HalPolicy>( |