diff options
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 |