diff options
Diffstat (limited to 'src/armnnUtils')
-rw-r--r-- | src/armnnUtils/DotSerializer.cpp | 4 | ||||
-rw-r--r-- | src/armnnUtils/DotSerializer.hpp | 6 |
2 files changed, 6 insertions, 4 deletions
diff --git a/src/armnnUtils/DotSerializer.cpp b/src/armnnUtils/DotSerializer.cpp index 1f36cf73ae..7416ff6bdf 100644 --- a/src/armnnUtils/DotSerializer.cpp +++ b/src/armnnUtils/DotSerializer.cpp @@ -116,7 +116,7 @@ DotAttributeSet & DotAttributeSet::AddAttribute(const std::string& name, const s return *this; } -DotEdge::DotEdge(std::ostream& stream, unsigned int fromNodeId, unsigned int toNodeId) +DotEdge::DotEdge(std::ostream& stream, LayerGuid fromNodeId, LayerGuid toNodeId) : DotBase(stream) { std::stringstream ss; @@ -176,7 +176,7 @@ NodeContent::~NodeContent() GetStream() << s; } -DotNode::DotNode(std::ostream& stream, unsigned int nodeId, const char* label) +DotNode::DotNode(std::ostream& stream, LayerGuid nodeId, const char* label) : DotBase(stream) { std::stringstream ss; diff --git a/src/armnnUtils/DotSerializer.hpp b/src/armnnUtils/DotSerializer.hpp index dfb8c7f22e..253fb768ec 100644 --- a/src/armnnUtils/DotSerializer.hpp +++ b/src/armnnUtils/DotSerializer.hpp @@ -5,6 +5,8 @@ #pragma once +#include <armnn/Types.hpp> + #include <ostream> #include <vector> #include <memory> @@ -75,7 +77,7 @@ private: class DotEdge : public DotBase { public: - explicit DotEdge(std::ostream& stream, unsigned int fromNodeId, unsigned int toNodeId); + explicit DotEdge(std::ostream& stream, LayerGuid fromNodeId, LayerGuid toNodeId); ~DotEdge(); DotAttributeSet& GetAttributeSet() { return *m_Attributes.get(); } @@ -99,7 +101,7 @@ private: class DotNode : public DotBase { public: - explicit DotNode(std::ostream& stream, unsigned int nodeId, const char* label); + explicit DotNode(std::ostream& stream, LayerGuid nodeId, const char* label); ~DotNode(); NodeContent& GetContents() { return *m_Contents.get(); } |