diff options
author | Keith Davis <keith.davis@arm.com> | 2022-10-19 14:53:05 +0100 |
---|---|---|
committer | Keith Davis <keith.davis@arm.com> | 2022-10-19 14:53:14 +0100 |
commit | f63b457c48fa2aeb01d8cff52475d23e5644f8f4 (patch) | |
tree | b2028a49e60412234020b878951df1e53d990baf /src/armnn | |
parent | 15f9c68adef324cd0158cea3d021c0f6bef5eecf (diff) | |
download | armnn-f63b457c48fa2aeb01d8cff52475d23e5644f8f4.tar.gz |
MLCE-545 INT8 TFLite model execution abnormal
* Bug fix where files were being overwritten at each debug layer
Signed-off-by: Keith Davis <keith.davis@arm.com>
Change-Id: I609fdc82afcee925824efb02183c7dbc942fced0
Diffstat (limited to 'src/armnn')
-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())); } |