diff options
Diffstat (limited to 'src/armnn/JsonPrinter.hpp')
-rw-r--r-- | src/armnn/JsonPrinter.hpp | 13 |
1 files changed, 12 insertions, 1 deletions
diff --git a/src/armnn/JsonPrinter.hpp b/src/armnn/JsonPrinter.hpp index 4af1609ee3..9e8ecb48e5 100644 --- a/src/armnn/JsonPrinter.hpp +++ b/src/armnn/JsonPrinter.hpp @@ -10,6 +10,7 @@ #include <set> #include <sstream> +#include <ProfilingGuid.hpp> #include "Instrument.hpp" #include "JsonUtils.hpp" @@ -27,7 +28,10 @@ struct JsonChildObject { // Object type changes according to the JsonObjectType specified in enum JsonChildObject(const std::string& label) - : m_Label(label), m_Unit(Measurement::Unit::TIME_MS), m_Type(JsonObjectType::Event) + : m_Label(label), + m_Unit(Measurement::Unit::TIME_MS), + m_Type(JsonObjectType::Event), + m_Guid(armnn::EmptyOptional()) {} JsonChildObject(const JsonChildObject&) = default; @@ -46,6 +50,11 @@ struct JsonChildObject } } + void SetGuid(profiling::ProfilingGuid guid) + { + m_Guid = Optional<profiling::ProfilingGuid>(guid); + } + void AddChild(const JsonChildObject& childObject) { m_Children.push_back(childObject); @@ -81,6 +90,7 @@ struct JsonChildObject std::string m_Label; Measurement::Unit m_Unit; JsonObjectType m_Type; + Optional<profiling::ProfilingGuid> m_Guid; std::vector<double> m_Measurements; std::vector<std::string> m_LayerDetailsList; std::vector<JsonChildObject> m_Children; @@ -96,6 +106,7 @@ public: void PrintLabel(const std::string& label, size_t id); void PrintUnit(armnn::Measurement::Unit unit); void PrintType(armnn::JsonObjectType type); + void PrintGuid(armnn::profiling::ProfilingGuid guid); void PrintMeasurementsList(const std::vector<double>& measurementsVector); public: |