aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/NEON/functions/NELSTMLayer.cpp
diff options
context:
space:
mode:
authorramelg01 <ramy.elgammal@arm.com>2021-09-17 17:36:57 +0100
committerramy.elgammal <ramy.elgammal@arm.com>2021-09-22 09:44:25 +0000
commitcbbb03813b79b7f0274b18436a78a79ff31e469e (patch)
treea80e529e419ddff1ef2f5fa72dfa37d29bf53977 /src/runtime/NEON/functions/NELSTMLayer.cpp
parent3ae3d88c1a305ef4fc0beed8fda3cfc39ddb2ae8 (diff)
downloadComputeLibrary-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/NELSTMLayer.cpp')
-rw-r--r--src/runtime/NEON/functions/NELSTMLayer.cpp8
1 files changed, 8 insertions, 0 deletions
diff --git a/src/runtime/NEON/functions/NELSTMLayer.cpp b/src/runtime/NEON/functions/NELSTMLayer.cpp
index 2eb5e9a61e..2d6be06499 100644
--- a/src/runtime/NEON/functions/NELSTMLayer.cpp
+++ b/src/runtime/NEON/functions/NELSTMLayer.cpp
@@ -29,6 +29,7 @@
#include "arm_compute/core/utils/misc/ShapeCalculator.h"
#include "arm_compute/core/utils/quantization/AsymmHelpers.h"
#include "arm_compute/runtime/common/LSTMParams.h"
+#include "src/common/utils/Log.h"
namespace arm_compute
{
@@ -67,6 +68,13 @@ void NELSTMLayer::configure(const ITensor *input,
forget_gate_bias, cell_bias, output_gate_bias,
output_state_in, cell_state_in,
scratch_buffer, output_state_out, cell_state_out, output);
+ ARM_COMPUTE_LOG_PARAMS(input,
+ input_to_forget_weights, input_to_cell_weights, input_to_output_weights,
+ recurrent_to_forget_weights, recurrent_to_cell_weights, recurrent_to_output_weights,
+ forget_gate_bias, cell_bias, output_gate_bias,
+ output_state_in, cell_state_in,
+ scratch_buffer, output_state_out, cell_state_out, output,
+ lstm_params, activation_info, cell_threshold, projection_threshold);
_is_layer_norm_lstm = lstm_params.use_layer_norm();