From e5339e7013cf24e5a34509fb0a60377e5f8a244e Mon Sep 17 00:00:00 2001
From: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Date: Wed, 28 Jul 2021 17:33:28 +0100
Subject: MLCE-530 Add support for UnidirectionalSequenceLstm to RefWorkload

 * Add implementation of IsUnidirectionalSequenceLstmSupported to RefLayerSupport
 * Add RefUnidirectionalSequenceLstmWorkload
 * Refactor Lstm to be able to use for Lstm and SequenceLstm
 * Unit tests

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ibc066d213213a11b955dfefbe518de643298ba0c
---
 src/backends/reference/workloads/Lstm.hpp | 61 +++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)
 create mode 100644 src/backends/reference/workloads/Lstm.hpp

(limited to 'src/backends/reference/workloads/Lstm.hpp')

diff --git a/src/backends/reference/workloads/Lstm.hpp b/src/backends/reference/workloads/Lstm.hpp
new file mode 100644
index 0000000000..7d0a1d436e
--- /dev/null
+++ b/src/backends/reference/workloads/Lstm.hpp
@@ -0,0 +1,61 @@
+//
+// Copyright © 2021 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include <armnn/TypesUtils.hpp>
+#include <backendsCommon/WorkloadData.hpp>
+
+#include "Encoders.hpp"
+#include "Decoders.hpp"
+
+namespace armnn
+{
+
+void LstmImpl(const LstmDescriptor& descriptor,
+              const TensorInfo& inputInfo,
+              const TensorInfo& outputInfo,
+              const TensorShape& inputToOutputWeightsShape,
+              const TensorShape& recurrentToOutputWeightsShape,
+              std::unique_ptr<Decoder<float>>& inputData,
+              std::unique_ptr<Decoder<float>>& outputStateIn,
+              std::unique_ptr<Decoder<float>>& cellStateIn,
+              std::unique_ptr<Encoder<float>>& outputStateOut,
+              std::unique_ptr<Encoder<float>>& cellStateOut,
+              std::unique_ptr<Encoder<float>>& output,
+              std::unique_ptr<Decoder<float>>& cellStateOutDecoder,
+              std::unique_ptr<Decoder<float>>& outputDecoder,
+              std::unique_ptr<Decoder<float>>& inputToInputWeightsTensor,
+              std::unique_ptr<Decoder<float>>& inputToForgetWeightsTensor,
+              std::unique_ptr<Decoder<float>>& inputToCellWeightsTensor,
+              std::unique_ptr<Decoder<float>>& inputToOutputWeightsTensor,
+              std::unique_ptr<Decoder<float>>& recurrentToInputWeightsTensor,
+              std::unique_ptr<Decoder<float>>& recurrentToForgetWeightsTensor,
+              std::unique_ptr<Decoder<float>>& recurrentToCellWeightsTensor,
+              std::unique_ptr<Decoder<float>>& recurrentToOutputWeightsTensor,
+              std::unique_ptr<Decoder<float>>& cellToInputWeightsTensor,
+              std::unique_ptr<Decoder<float>>& cellToForgetWeightsTensor,
+              std::unique_ptr<Decoder<float>>& cellToOutputWeightsTensor,
+              std::unique_ptr<Decoder<float>>& inputGateBiasTensor,
+              std::unique_ptr<Decoder<float>>& forgetGateBiasTensor,
+              std::unique_ptr<Decoder<float>>& cellBiasTensor,
+              std::unique_ptr<Decoder<float>>& outputGateBiasTensor,
+              std::unique_ptr<Decoder<float>>& projectionWeightsTensor,
+              std::unique_ptr<Decoder<float>>& projectionBiasTensor,
+              std::unique_ptr<Decoder<float>>& inputLayerNormWeights,
+              std::unique_ptr<Decoder<float>>& forgetLayerNormWeights,
+              std::unique_ptr<Decoder<float>>& cellLayerNormWeights,
+              std::unique_ptr<Decoder<float>>& outputLayerNormWeights,
+              std::unique_ptr<Encoder<float>>& inputGateScratch,
+              std::unique_ptr<Encoder<float>>& cellScratch,
+              std::unique_ptr<Encoder<float>>& forgetGateScratch,
+              std::unique_ptr<Encoder<float>>& outputGateScratch,
+              std::unique_ptr<Decoder<float>>& inputGateScratchDecoder,
+              std::unique_ptr<Decoder<float>>& cellScratchDecoder,
+              std::unique_ptr<Decoder<float>>& forgetGateScratchDecoder,
+              std::unique_ptr<Decoder<float>>& outputGateScratchDecoder,
+              float layerNormEpsilon);
+
+} //namespace armnn
-- 
cgit v1.2.1