From 09daf4ddf5940d18ce95e7dd0859d1dace3b133e Mon Sep 17 00:00:00 2001 From: Michalis Spyrou Date: Thu, 28 Jun 2018 17:07:22 +0100 Subject: COMPMID-1246 Change CLLSTM in order to match android tests Allow cell to input weights to be nullptr if CIFG and peephole are both enabled. Change-Id: I6df705d69551f0fddeedd41b2044278d4575469c Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/137902 Tested-by: Jenkins Reviewed-by: Georgios Pinitas --- arm_compute/runtime/CL/functions/CLLSTMLayer.h | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) (limited to 'arm_compute/runtime/CL/functions/CLLSTMLayer.h') diff --git a/arm_compute/runtime/CL/functions/CLLSTMLayer.h b/arm_compute/runtime/CL/functions/CLLSTMLayer.h index cf47f34290..bed1fa9f56 100644 --- a/arm_compute/runtime/CL/functions/CLLSTMLayer.h +++ b/arm_compute/runtime/CL/functions/CLLSTMLayer.h @@ -96,15 +96,13 @@ public: } /** Set peephole tensor parameters. * - * @param[in] cell_to_input_weights 1D weights tensor with dimensions [num_units]. Data type supported: Data types supported: F16/F32. - * @param[in] cell_to_forget_weights 1D weights tensor with dimensions [num_units]. Data type supported: Same as @p cell_to_input_weights. + * @param[in] cell_to_forget_weights 1D weights tensor with dimensions [num_units]. Data type supported: Data types supported: F16/F32. * @param[in] cell_to_output_weights 1D weights tensor with dimensions [num_units]. Data type supported: Same as @p cell_to_input_weights. * * @return Reference to this LSTMParams object */ - LSTMParams &set_peephole_params(const T *cell_to_input_weights, const T *cell_to_forget_weights, const T *cell_to_output_weights) + LSTMParams &set_peephole_params(const T *cell_to_forget_weights, const T *cell_to_output_weights) { - _cell_to_input_weights = cell_to_input_weights; _cell_to_forget_weights = cell_to_forget_weights; _cell_to_output_weights = cell_to_output_weights; _has_peephole_opt = true; -- cgit v1.2.1