diff options
Diffstat (limited to 'src/runtime/NEON/functions/NELSTMLayer.cpp')
| -rw-r--r-- | src/runtime/NEON/functions/NELSTMLayer.cpp | 8 |
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(); |