diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2017-10-02 17:44:52 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
commit | e472082f831815c217677e3f1802ecaae1348e65 (patch) | |
tree | 896fc2631e0d1113f9bd33ffb730973e886f2717 /src/graph/nodes/FullyConnectedLayer.cpp | |
parent | 744b5edd1e7eedab8ac52a8cea33bf62fb95affc (diff) | |
download | ComputeLibrary-e472082f831815c217677e3f1802ecaae1348e65.tar.gz |
COMPMID-549 Create a Logger for GraphAPI
Change-Id: If912d8232e12cd496923d55d386898450dac09e2
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/89897
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 | 30 |
1 files changed, 10 insertions, 20 deletions
diff --git a/src/graph/nodes/FullyConnectedLayer.cpp b/src/graph/nodes/FullyConnectedLayer.cpp index c317660b20..6b21810a36 100644 --- a/src/graph/nodes/FullyConnectedLayer.cpp +++ b/src/graph/nodes/FullyConnectedLayer.cpp @@ -24,6 +24,7 @@ #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" @@ -112,35 +113,24 @@ std::unique_ptr<arm_compute::IFunction> FullyConnectedLayer::instantiate_node(Gr std::unique_ptr<arm_compute::IFunction> func; _target_hint = ctx.hints().target_hint(); - _input = input; - _output = output; if(_target_hint == TargetHint::OPENCL) { func = instantiate<TargetHint::OPENCL>(input, _weights, _biases, output); + ARM_COMPUTE_LOG("Instantiating CLFullyConnectedLayer"); } else { func = instantiate<TargetHint::NEON>(input, _weights, _biases, output); + ARM_COMPUTE_LOG("Instantiating NEFullyConnectedLayer"); } - return func; -} + ARM_COMPUTE_LOG(" Type: " << input->info()->data_type() + << " Input Shape: " << input->info()->tensor_shape() + << " Weights shape: " << _weights.info().tensor_shape() + << " Biases Shape: " << _biases.info().tensor_shape() + << " Output Shape: " << output->info()->tensor_shape() + << std::endl); -void FullyConnectedLayer::print_info() -{ - if(_target_hint == TargetHint::OPENCL) - { - std::cout << "Instantiating CLFullyConnectedLayer"; - } - else - { - std::cout << "Instantiating NEFullyConnectedLayer"; - } - std::cout << " Type: " << _input->info()->data_type() - << " Input Shape: " << _input->info()->tensor_shape() - << " Weights shape: " << _weights.info().tensor_shape() - << " Biases Shape: " << _biases.info().tensor_shape() - << " Output Shape: " << _output->info()->tensor_shape() - << std::endl; + return func; } |