diff options
| author | ramelg01 <ramy.elgammal@arm.com> | 2021-09-17 17:36:57 +0100 |
|---|---|---|
| committer | ramy.elgammal <ramy.elgammal@arm.com> | 2021-09-22 09:44:25 +0000 |
| commit | cbbb03813b79b7f0274b18436a78a79ff31e469e (patch) | |
| tree | a80e529e419ddff1ef2f5fa72dfa37d29bf53977 /src/runtime/NEON/functions/NELSTMLayerQuantized.cpp | |
| parent | 3ae3d88c1a305ef4fc0beed8fda3cfc39ddb2ae8 (diff) | |
| download | ComputeLibrary-cbbb03813b79b7f0274b18436a78a79ff31e469e.tar.gz | |
Provide logging for configure functions in all NEON functions
Partially Resolves: COMPMID-4718
Signed-off-by: Ramy Elgammal <ramy.elgammal@arm.com>
Change-Id: I655268c57fa126d9c99981c49d345a3aac75646e
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6286
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Marquez Tello <pablo.tello@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>
Diffstat (limited to 'src/runtime/NEON/functions/NELSTMLayerQuantized.cpp')
| -rw-r--r-- | src/runtime/NEON/functions/NELSTMLayerQuantized.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/runtime/NEON/functions/NELSTMLayerQuantized.cpp b/src/runtime/NEON/functions/NELSTMLayerQuantized.cpp index eb7d584da2..cfdeb000e0 100644 --- a/src/runtime/NEON/functions/NELSTMLayerQuantized.cpp +++ b/src/runtime/NEON/functions/NELSTMLayerQuantized.cpp @@ -26,6 +26,7 @@ #include "arm_compute/core/Utils.h" #include "arm_compute/core/Validate.h" #include "arm_compute/core/utils/quantization/AsymmHelpers.h" +#include "src/common/utils/Log.h" #include "src/core/helpers/AutoConfiguration.h" #include <cmath> @@ -72,6 +73,10 @@ void NELSTMLayerQuantized::configure(const ITensor *input, recurrent_to_input_weights->info(), recurrent_to_forget_weights->info(), recurrent_to_cell_weights->info(), recurrent_to_output_weights->info(), input_gate_bias->info(), forget_gate_bias->info(), cell_bias->info(), output_gate_bias->info(), cell_state_in->info(), output_state_in->info(), cell_state_out->info(), output_state_out->info())); + ARM_COMPUTE_LOG_PARAMS(input, input_to_input_weights, input_to_forget_weights, input_to_cell_weights, input_to_output_weights, + recurrent_to_input_weights, recurrent_to_forget_weights, recurrent_to_cell_weights, recurrent_to_output_weights, + input_gate_bias, forget_gate_bias, cell_bias, output_gate_bias, cell_state_in, output_state_in, cell_state_out, output_state_out); + const int input_size = input->info()->dimension(0); const int batch_size = input->info()->dimension(1); const int output_size = input_to_input_weights->info()->dimension(1); |