diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2017-10-12 17:34:20 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
| commit | 7d3d1b923b7793f1cf5e29c78bfda2582522cf25 (patch) | |
| tree | 7a924958ee7f4c9b7e6245170513862dcc016b87 /src/graph/nodes/FullyConnectedLayer.cpp | |
| parent | 3756186e54d77639564e999082f4bbd1ceec5a2f (diff) | |
| download | ComputeLibrary-7d3d1b923b7793f1cf5e29c78bfda2582522cf25.tar.gz | |
COMPMID-619: Logging interface
Change-Id: I73d1433ee7a682aeabb7540aa2ea1f6564f90aae
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/91775
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'src/graph/nodes/FullyConnectedLayer.cpp')
| -rw-r--r-- | src/graph/nodes/FullyConnectedLayer.cpp | 15 |
1 files changed, 8 insertions, 7 deletions
diff --git a/src/graph/nodes/FullyConnectedLayer.cpp b/src/graph/nodes/FullyConnectedLayer.cpp index fa5ead8bdd..5f4807ad48 100644 --- a/src/graph/nodes/FullyConnectedLayer.cpp +++ b/src/graph/nodes/FullyConnectedLayer.cpp @@ -24,7 +24,6 @@ #include "arm_compute/graph/nodes/FullyConnectedLayer.h" #include "arm_compute/core/Helpers.h" -#include "arm_compute/core/Logger.h" #include "arm_compute/runtime/CL/functions/CLFullyConnectedLayer.h" #include "arm_compute/runtime/NEON/functions/NEFullyConnectedLayer.h" #include "support/ToolchainSupport.h" @@ -123,18 +122,20 @@ std::unique_ptr<arm_compute::IFunction> FullyConnectedLayer::instantiate_node(Gr if(_target_hint == TargetHint::OPENCL) { func = instantiate<TargetHint::OPENCL>(in, _weights, _biases, out); + ARM_COMPUTE_LOG_GRAPH_INFO("Instantiating CLFullyConnectedLayer"); } else { func = instantiate<TargetHint::NEON>(in, _weights, _biases, out); + ARM_COMPUTE_LOG_GRAPH_INFO("Instantiating NEFullyConnectedLayer"); } - ARM_COMPUTE_LOG(" Type: " << in->info()->data_type() - << " Input Shape: " << in->info()->tensor_shape() - << " Weights shape: " << _weights.info().tensor_shape() - << " Biases Shape: " << _biases.info().tensor_shape() - << " Output Shape: " << out->info()->tensor_shape() - << std::endl); + ARM_COMPUTE_LOG_GRAPH_INFO(" Type: " << in->info()->data_type() + << " Input Shape: " << in->info()->tensor_shape() + << " Weights shape: " << _weights.info().tensor_shape() + << " Biases Shape: " << _biases.info().tensor_shape() + << " Output Shape: " << out->info()->tensor_shape() + << std::endl); return func; } |