diff options
Diffstat (limited to 'src/cpu/operators/CpuConv2d.cpp')
| -rw-r--r-- | src/cpu/operators/CpuConv2d.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/cpu/operators/CpuConv2d.cpp b/src/cpu/operators/CpuConv2d.cpp index 3878e0de58..fa8a7a185c 100644 --- a/src/cpu/operators/CpuConv2d.cpp +++ b/src/cpu/operators/CpuConv2d.cpp @@ -24,6 +24,7 @@ #include "src/cpu/operators/CpuConv2d.h" #include "arm_compute/runtime/NEON/NEScheduler.h" #include "arm_compute/runtime/NEON/functions/NEFFTConvolutionLayer.h" +#include "src/common/utils/Log.h" #include "src/cpu/operators/CpuDirectConv2d.h" #include "src/cpu/operators/CpuGemm.h" #include "src/cpu/operators/CpuGemmConv2d.h" @@ -50,6 +51,8 @@ void CpuConv2d::configure(ITensorInfo *input, ITensorInfo *weights, const ITenso ARM_COMPUTE_ERROR_THROW_ON(CpuConv2d::validate(input, weights, biases, output, conv_info, weights_info, dilation, act_info, enable_fast_math, num_groups)); + ARM_COMPUTE_LOG_PARAMS(input, weights, biases, output, conv_info, weights_info, dilation, act_info, enable_fast_math, num_groups); + const Conv2dInfo info(conv_info, dilation, act_info, enable_fast_math, num_groups); switch(CpuConv2d::get_convolution_method(input, weights, output, conv_info, weights_info, dilation, act_info, enable_fast_math)) { |