diff options
author | keidav01 <keith.davis@arm.com> | 2018-12-11 16:14:20 +0000 |
---|---|---|
committer | Keith Davis Arm <keith.davis@arm.com> | 2018-12-11 17:11:20 +0000 |
commit | 738c2e6a647b886750e1bc3daa6dd615a0244baa (patch) | |
tree | ef88c6a5ee1076471bd36b85c30afb36a68bbfc9 /src/armnn/NetworkUtils.hpp | |
parent | 44a7167e0f13dc1d703cd573f57636fde711c618 (diff) | |
download | armnn-738c2e6a647b886750e1bc3daa6dd615a0244baa.tar.gz |
IVGCVSW-1434 Add debug mode to Optimizer
* Modified optimizer to support debug mode via DebugLayer
Change-Id: Ic8f313778e55540c182cf99876c44a0823be04c6
Diffstat (limited to 'src/armnn/NetworkUtils.hpp')
-rw-r--r-- | src/armnn/NetworkUtils.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnn/NetworkUtils.hpp b/src/armnn/NetworkUtils.hpp index dbb85380ff..b81d5cb5e7 100644 --- a/src/armnn/NetworkUtils.hpp +++ b/src/armnn/NetworkUtils.hpp @@ -14,4 +14,6 @@ std::vector<ConvertFp16ToFp32Layer*> InsertConvertFp16ToFp32LayersBefore(Graph& std::vector<ConvertFp32ToFp16Layer*> InsertConvertFp32ToFp16LayersAfter(Graph& graph, Layer& layer); +std::vector<DebugLayer*> InsertDebugLayerAfter(Graph& graph, Layer& layer); + } // namespace armnn |