diff options
| author | Michalis Spyrou <michalis.spyrou@arm.com> | 2018-03-28 14:35:08 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:49:16 +0000 |
| commit | 3672df36e7b61f280d66b6cbb11c2b4254fbbc8c (patch) | |
| tree | 98949c82fc856c5ac9dd43f05a59d712f3e024d3 /src/runtime/CL/functions/CLRNNLayer.cpp | |
| parent | 9a8c672dd6eb21448fbfb4b636104323d128de88 (diff) | |
| download | ComputeLibrary-3672df36e7b61f280d66b6cbb11c2b4254fbbc8c.tar.gz | |
COMPMID-959 Add missing validate check in CLRNNLayer
Change-Id: I9346e62dc4eef4f3dc7a70160ac9370f4deeb8fa
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/126185
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/runtime/CL/functions/CLRNNLayer.cpp')
| -rw-r--r-- | src/runtime/CL/functions/CLRNNLayer.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/runtime/CL/functions/CLRNNLayer.cpp b/src/runtime/CL/functions/CLRNNLayer.cpp index 75eac0959f..4843ba6364 100644 --- a/src/runtime/CL/functions/CLRNNLayer.cpp +++ b/src/runtime/CL/functions/CLRNNLayer.cpp @@ -68,6 +68,9 @@ Status CLRNNLayer::validate(const ITensorInfo *input, const ITensorInfo *weights void CLRNNLayer::configure(const ICLTensor *input, const ICLTensor *weights, const ICLTensor *recurrent_weights, const ICLTensor *bias, ICLTensor *hidden_state, ICLTensor *output, ActivationLayerInfo &info) { + ARM_COMPUTE_ERROR_ON_NULLPTR(input, weights, recurrent_weights, bias, hidden_state, output); + ARM_COMPUTE_ERROR_THROW_ON(CLRNNLayer::validate(input->info(), weights->info(), recurrent_weights->info(), bias->info(), hidden_state->info(), output->info(), info)); + const int idx_height = get_data_layout_dimension_index(input->info()->data_layout(), DataLayoutDimension::HEIGHT); TensorShape shape = compute_rnn_shape(recurrent_weights->info(), hidden_state->info()->dimension(idx_height)); |