aboutsummaryrefslogtreecommitdiff
path: root/src/graph/nodes/SoftmaxLayer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/graph/nodes/SoftmaxLayer.cpp')
-rw-r--r--src/graph/nodes/SoftmaxLayer.cpp11
1 files changed, 6 insertions, 5 deletions
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<arm_compute::IFunction> SoftmaxLayer::instantiate_node(GraphCont
if(_target_hint == TargetHint::OPENCL)
{
func = instantiate<TargetHint::OPENCL>(in, out);
+ ARM_COMPUTE_LOG_GRAPH_INFO("Instantiating CLSoftmaxLayer");
}
else
{
func = instantiate<TargetHint::NEON>(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;
}