diff options
Diffstat (limited to 'src/armnn/layers/DebugLayer.hpp')
-rw-r--r-- | src/armnn/layers/DebugLayer.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/armnn/layers/DebugLayer.hpp b/src/armnn/layers/DebugLayer.hpp index fe7ad5c9e5..662195d7bc 100644 --- a/src/armnn/layers/DebugLayer.hpp +++ b/src/armnn/layers/DebugLayer.hpp @@ -34,9 +34,13 @@ protected: /// Constructor to create a DebugLayer. /// @param [in] name Optional name for the layer. DebugLayer(const char* name); + DebugLayer(const char* name, bool toFile); /// Default destructor ~DebugLayer() = default; + +private: + bool m_ToFile; }; } // namespace armnn |