diff options
author | Michalis Spyrou <michalis.spyrou@arm.com> | 2018-06-05 11:45:48 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:53:09 +0000 |
commit | 542e92d95536f2ab7fc6f1cc1aa1bd4f1d471212 (patch) | |
tree | a4c03807d9731c1305b6f446282d3f4b97cfb595 /tests/datasets | |
parent | 72219330fd85b1271e714d4ba894d6d8e26340c9 (diff) | |
download | ComputeLibrary-542e92d95536f2ab7fc6f1cc1aa1bd4f1d471212.tar.gz |
COMPMID-1067 NEON RNN FP32 / FP16
Change-Id: I440df2b2af512fd874651baf28428caa6f8e0b41
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/134433
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'tests/datasets')
-rw-r--r-- | tests/datasets/RNNLayerDataset.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/datasets/RNNLayerDataset.h b/tests/datasets/RNNLayerDataset.h index 616a69e213..40d1b934f3 100644 --- a/tests/datasets/RNNLayerDataset.h +++ b/tests/datasets/RNNLayerDataset.h @@ -131,7 +131,7 @@ class SmallRNNLayerDataset final : public RNNLayerDataset public: SmallRNNLayerDataset() { - add_config(TensorShape(8U, 2U), TensorShape(8U, 16U), TensorShape(16U, 16U), TensorShape(16U), TensorShape(16U, 2U), ActivationLayerInfo()); + add_config(TensorShape(128U, 16U), TensorShape(128U, 32U), TensorShape(32U, 32U), TensorShape(32U), TensorShape(32U, 16U), ActivationLayerInfo()); } }; |