diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2017-10-02 17:44:52 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
commit | e472082f831815c217677e3f1802ecaae1348e65 (patch) | |
tree | 896fc2631e0d1113f9bd33ffb730973e886f2717 /examples | |
parent | 744b5edd1e7eedab8ac52a8cea33bf62fb95affc (diff) | |
download | ComputeLibrary-e472082f831815c217677e3f1802ecaae1348e65.tar.gz |
COMPMID-549 Create a Logger for GraphAPI
Change-Id: If912d8232e12cd496923d55d386898450dac09e2
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/89897
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'examples')
-rw-r--r-- | examples/graph_alexnet.cpp | 3 | ||||
-rw-r--r-- | examples/graph_lenet.cpp | 3 |
2 files changed, 4 insertions, 2 deletions
diff --git a/examples/graph_alexnet.cpp b/examples/graph_alexnet.cpp index 9c736c5df1..dce7132785 100644 --- a/examples/graph_alexnet.cpp +++ b/examples/graph_alexnet.cpp @@ -25,6 +25,7 @@ #error "This example needs to be built with -DARM_COMPUTE_CL" #endif /* ARM_COMPUTE_CL */ +#include "arm_compute/core/Logger.h" #include "arm_compute/graph/Graph.h" #include "arm_compute/graph/Nodes.h" #include "arm_compute/runtime/CL/CLScheduler.h" @@ -102,7 +103,7 @@ void main_graph_alexnet(int argc, const char **argv) } Graph graph; - graph.set_info_enablement(true); + arm_compute::Logger::get().set_logger(std::cout, arm_compute::LoggerVerbosity::INFO); graph << hint << Tensor(TensorInfo(TensorShape(227U, 227U, 3U, batches), 1, DataType::F32), DummyAccessor()) diff --git a/examples/graph_lenet.cpp b/examples/graph_lenet.cpp index 51b0881b1b..1427abe15f 100644 --- a/examples/graph_lenet.cpp +++ b/examples/graph_lenet.cpp @@ -25,6 +25,7 @@ #error "This example needs to be built with -DARM_COMPUTE_CL" #endif /* ARM_COMPUTE_CL */ +#include "arm_compute/core/Logger.h" #include "arm_compute/graph/Graph.h" #include "arm_compute/graph/Nodes.h" #include "arm_compute/runtime/CL/CLScheduler.h" @@ -101,7 +102,7 @@ void main_graph_lenet(int argc, const char **argv) } Graph graph; - graph.set_info_enablement(true); + arm_compute::Logger::get().set_logger(std::cout, arm_compute::LoggerVerbosity::INFO); //conv1 << pool1 << conv2 << pool2 << fc1 << act1 << fc2 << smx graph << hint |