diff options
Diffstat (limited to 'src/runtime/CL/functions/CLLSTMLayer.cpp')
| -rw-r--r-- | src/runtime/CL/functions/CLLSTMLayer.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/runtime/CL/functions/CLLSTMLayer.cpp b/src/runtime/CL/functions/CLLSTMLayer.cpp index 245016efcb..a89c4e3dbf 100644 --- a/src/runtime/CL/functions/CLLSTMLayer.cpp +++ b/src/runtime/CL/functions/CLLSTMLayer.cpp @@ -282,7 +282,7 @@ void CLLSTMLayer::configure(const ICLTensor *input, // Vector for holding the tensors to store in scratch buffer std::vector<ICLTensor *> scratch_inputs; - if(lstm_params.has_cifg_opt()) + if(!lstm_params.has_cifg_opt()) { scratch_inputs.emplace_back(&_input_gate_out1); } @@ -444,7 +444,7 @@ Status CLLSTMLayer::validate(const ITensorInfo *input, // Validate scratch concatenation std::vector<ITensorInfo *> inputs_vector_info_raw; - if(lstm_params.has_cifg_opt()) + if(!lstm_params.has_cifg_opt()) { inputs_vector_info_raw.push_back(&input_gate); } |