From 7d3d1b923b7793f1cf5e29c78bfda2582522cf25 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Thu, 12 Oct 2017 17:34:20 +0100 Subject: COMPMID-619: Logging interface Change-Id: I73d1433ee7a682aeabb7540aa2ea1f6564f90aae Reviewed-on: http://mpd-gerrit.cambridge.arm.com/91775 Tested-by: Kaizen Reviewed-by: Anthony Barbier --- src/graph/nodes/SoftmaxLayer.cpp | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) (limited to 'src/graph/nodes/SoftmaxLayer.cpp') diff --git a/src/graph/nodes/SoftmaxLayer.cpp b/src/graph/nodes/SoftmaxLayer.cpp index e3345f1400..3cdbc9c96a 100644 --- a/src/graph/nodes/SoftmaxLayer.cpp +++ b/src/graph/nodes/SoftmaxLayer.cpp @@ -23,7 +23,6 @@ */ #include "arm_compute/graph/nodes/SoftmaxLayer.h" -#include "arm_compute/core/Logger.h" #include "arm_compute/runtime/CL/CLTensor.h" #include "arm_compute/runtime/CL/functions/CLSoftmaxLayer.h" #include "arm_compute/runtime/NEON/functions/NESoftmaxLayer.h" @@ -76,16 +75,18 @@ std::unique_ptr SoftmaxLayer::instantiate_node(GraphCont if(_target_hint == TargetHint::OPENCL) { func = instantiate(in, out); + ARM_COMPUTE_LOG_GRAPH_INFO("Instantiating CLSoftmaxLayer"); } else { func = instantiate(in, out); + ARM_COMPUTE_LOG_GRAPH_INFO("Instantiating NESoftmaxLayer"); } - ARM_COMPUTE_LOG(" Data Type: " << in->info()->data_type() - << " Input shape: " << in->info()->tensor_shape() - << " Output shape: " << out->info()->tensor_shape() - << std::endl); + ARM_COMPUTE_LOG_GRAPH_INFO(" Data Type: " << in->info()->data_type() + << " Input shape: " << in->info()->tensor_shape() + << " Output shape: " << out->info()->tensor_shape() + << std::endl); return func; } -- cgit v1.2.1