commit bf8b01dfbfdca124673ade33c5eac8f3748d7abd
author Kaizen <kaizen@arm.com> Thu Oct 12 14:26:51 2017 +0100
committer Anthony Barbier <anthony.barbier@arm.com> Thu Oct 12 15:42:58 2017 +0100
tree 9504f134ca292b988e7065b22229e9bbe0493d00
parent 8938bd3f40ea62ff56d6ed4e2db0a8aee34dd64a

arm_compute v17.10

Change-Id: If1489af40eccd0219ede8946577afbf04db31b29

examples/graph_lenet.cpp

diff --git a/examples/graph_lenet.cpp b/examples/graph_lenet.cpp
index 676fdb9..1427abe 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"
@@ -93,16 +94,18 @@
     }
 
     // Check if OpenCL is available and initialize the scheduler
+    TargetHint hint = TargetHint::NEON;
     if(arm_compute::opencl_is_available())
     {
         arm_compute::CLScheduler::get().default_init();
+        hint = TargetHint::OPENCL;
     }
 
     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::OPENCL
+    graph << hint
           << Tensor(TensorInfo(TensorShape(28U, 28U, 1U, batches), 1, DataType::F32), DummyAccessor())
           << ConvolutionLayer(
               5U, 5U, 20U,