diff options
Diffstat (limited to 'src/armnn/Profiling.cpp')
| -rw-r--r-- | src/armnn/Profiling.cpp | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/src/armnn/Profiling.cpp b/src/armnn/Profiling.cpp index 6a3522ba29..e7df459103 100644 --- a/src/armnn/Profiling.cpp +++ b/src/armnn/Profiling.cpp @@ -411,7 +411,12 @@ void ProfilerImpl::Print(std::ostream& outStream) const || m_DetailsToStdOutMethod == ProfilingDetailsMethod::DetailsWithEvents)) { JsonChildObject detailsObject{ "layer_details" }; - ConfigureDetailsObject(detailsObject, m_ProfilingDetails.get()->GetProfilingDetails()); + if (m_DetailsToStdOutMethod == ProfilingDetailsMethod::DetailsOnly) + { + detailsObject.EnableDetailsOnly(); + } + detailsObject.SetType(JsonObjectType::ExecObjectDesc); + detailsObject.SetAndParseDetails(m_ProfilingDetails.get()->GetProfilingDetails()); size_t id = 0; printer.PrintJsonChildObject(detailsObject, id); @@ -428,9 +433,9 @@ void ProfilerImpl::Print(std::ostream& outStream) const printer.PrintSeparator(); printer.PrintNewLine(); printer.PrintJsonChildObject(inferenceObject, id); + printer.PrintNewLine(); } // end of ArmNN - printer.PrintNewLine(); printer.PrintFooter(); // end of main JSON object |