diff options
Diffstat (limited to 'src/runtime/CL/functions/CLLSTMLayer.cpp')
| -rw-r--r-- | src/runtime/CL/functions/CLLSTMLayer.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/runtime/CL/functions/CLLSTMLayer.cpp b/src/runtime/CL/functions/CLLSTMLayer.cpp index 0122162073..9f17a52812 100644 --- a/src/runtime/CL/functions/CLLSTMLayer.cpp +++ b/src/runtime/CL/functions/CLLSTMLayer.cpp @@ -32,6 +32,8 @@ #include "src/core/CL/kernels/CLFillBorderKernel.h" #include "src/gpu/cl/kernels/ClTransposeKernel.h" +#include "src/common/utils/Log.h" + namespace arm_compute { using namespace arm_compute::misc::shape_calculator; @@ -83,6 +85,10 @@ void CLLSTMLayer::configure(const CLCompileContext &compile_context, const ICLTe 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(); // Set lstm parameters |