diff options
Diffstat (limited to 'src/armnn/Network.cpp')
-rw-r--r-- | src/armnn/Network.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index bb6eb19fa3..d3ce7ab62c 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -11,6 +11,7 @@ #include "SubgraphViewSelector.hpp" #include "BackendSettings.hpp" #include "optimizations/All.hpp" +#include "armnnUtils/Filesystem.hpp" #include <armnn/backends/TensorHandle.hpp> #include <armnn/backends/WorkloadFactory.hpp> @@ -1816,6 +1817,8 @@ IOptimizedNetworkPtr Optimize(const Graph& inGraph, } else if (options.m_DebugToFile) { + // Setup the output file path + armnnUtils::Filesystem::CreateDirectory("/ArmNNIntermediateLayerOutputs"); Optimizer::Pass(optGraph, MakeOptimizations(InsertDebugToFileLayer())); } |