diff options
Diffstat (limited to 'src/graph/nodes/PoolingLayer.cpp')
-rw-r--r-- | src/graph/nodes/PoolingLayer.cpp | 27 |
1 files changed, 8 insertions, 19 deletions
diff --git a/src/graph/nodes/PoolingLayer.cpp b/src/graph/nodes/PoolingLayer.cpp index 2a5e4cb3d8..317cf4d14f 100644 --- a/src/graph/nodes/PoolingLayer.cpp +++ b/src/graph/nodes/PoolingLayer.cpp @@ -23,6 +23,7 @@ */ #include "arm_compute/graph/nodes/PoolingLayer.h" +#include "arm_compute/core/Logger.h" #include "arm_compute/runtime/CL/CLTensor.h" #include "arm_compute/runtime/CL/functions/CLPoolingLayer.h" #include "arm_compute/runtime/NEON/functions/NEPoolingLayer.h" @@ -71,34 +72,22 @@ std::unique_ptr<arm_compute::IFunction> PoolingLayer::instantiate_node(GraphCont { 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, output, _pool_info); + ARM_COMPUTE_LOG("Instantiating CLPoolingLayer"); } else { func = instantiate<TargetHint::NEON>(input, output, _pool_info); + ARM_COMPUTE_LOG("Instantiating NEPoolingLayer"); } - return func; -} + ARM_COMPUTE_LOG(" Data Type: " << input->info()->data_type() + << " Input shape: " << input->info()->tensor_shape() + << " Output shape: " << output->info()->tensor_shape() + << " Pooling info: " << _pool_info << std::endl); -void PoolingLayer::print_info() -{ - if(_target_hint == TargetHint::OPENCL) - { - std::cout << "Instantiating CLPoolingLayer"; - } - else - { - std::cout << "Instantiating NEPoolingLayer"; - } - - std::cout << " Data Type: " << _input->info()->data_type() - << " Input shape: " << _input->info()->tensor_shape() - << " Output shape: " << _output->info()->tensor_shape() - << " Pooling info: " << _pool_info << std::endl; + return func; } |