aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/test/QLstmEndToEndTestImpl.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/backendsCommon/test/QLstmEndToEndTestImpl.cpp')
-rw-r--r--src/backends/backendsCommon/test/QLstmEndToEndTestImpl.cpp21
1 files changed, 14 insertions, 7 deletions
diff --git a/src/backends/backendsCommon/test/QLstmEndToEndTestImpl.cpp b/src/backends/backendsCommon/test/QLstmEndToEndTestImpl.cpp
index e2147fc59b..7c87f358d6 100644
--- a/src/backends/backendsCommon/test/QLstmEndToEndTestImpl.cpp
+++ b/src/backends/backendsCommon/test/QLstmEndToEndTestImpl.cpp
@@ -80,22 +80,26 @@ void QLstmEndToEnd(const std::vector<armnn::BackendId>& backends)
const armnn::TensorInfo inputWeightsInfo({outputSize, inputSize},
armnn::DataType::QSymmS8,
weightsScale,
- weightsOffset, true);
+ weightsOffset,
+ true);
const armnn::TensorInfo recurrentWeightsInfo({outputSize, outputSize},
armnn::DataType::QSymmS8,
weightsScale,
- weightsOffset, true);
+ weightsOffset,
+ true);
const armnn::TensorInfo biasInfo({outputSize},
armnn::DataType::Signed32,
biasScale,
- biasOffset, true);
+ biasOffset,
+ true);
const armnn::TensorInfo layerNormWeightsInfo({numUnits},
armnn::DataType::QSymmS16,
layerNormScale,
- layerNormOffset, true);
+ layerNormOffset,
+ true);
// Mandatory params
const std::vector<int8_t> inputToForgetWeightsVector =
@@ -179,17 +183,20 @@ void QLstmEndToEnd(const std::vector<armnn::BackendId>& backends)
const armnn::TensorInfo inputInfo({numBatches , inputSize},
armnn::DataType::QAsymmS8,
inputScale,
- inputOffset, true);
+ inputOffset,
+ true);
const armnn::TensorInfo cellStateInfo({numBatches , numUnits},
armnn::DataType::QSymmS16,
cellStateScale,
- cellStateOffset, true);
+ cellStateOffset,
+ true);
const armnn::TensorInfo outputStateInfo({numBatches , outputSize},
armnn::DataType::QAsymmS8,
outputScale,
- outputOffset, true);
+ outputOffset,
+ true);
// Input tensor data
const std::vector<int8_t> inputVector = {90, 102, 13, 26, 38, 102, 13, 26, 51, 64};