diff options
author | Francis Murtagh <francis.murtagh@arm.com> | 2019-08-14 09:51:36 +0100 |
---|---|---|
committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-08-14 10:37:35 +0000 |
commit | bb590b42b6a877e7caf0c5e73070bab42f44c760 (patch) | |
tree | 99b82801365a5fb25ea17725755244fb5f6f9692 /src/armnnSerializer | |
parent | d65cb800d2c5acca3f31e9358fa5bfbe153e3aa9 (diff) | |
download | armnn-bb590b42b6a877e7caf0c5e73070bab42f44c760.tar.gz |
IVGCVSW-3474 Refactor Lstm and QuantizedLstm Param Getters
* Change Getter Signatures to follow coding guidelines
Change-Id: Ic02621e834dbf79b9df63f8b4c6339f71651e944
Signed-off-by: Francis Murtagh <francis.murtagh@arm.com>
Diffstat (limited to 'src/armnnSerializer')
-rw-r--r-- | src/armnnSerializer/Serializer.cpp | 28 |
1 files changed, 14 insertions, 14 deletions
diff --git a/src/armnnSerializer/Serializer.cpp b/src/armnnSerializer/Serializer.cpp index af4dc7a926..d35be6f6d8 100644 --- a/src/armnnSerializer/Serializer.cpp +++ b/src/armnnSerializer/Serializer.cpp @@ -1049,20 +1049,20 @@ void SerializerVisitor::VisitQuantizedLstmLayer(const armnn::IConnectableLayer* auto fbQuantizedLstmBaseLayer = CreateLayerBase(layer, serializer::LayerType::LayerType_QuantizedLstm); // Get input parameters - auto inputToInputWeights = CreateConstTensorInfo(params.get_InputToInputWeights()); - auto inputToForgetWeights = CreateConstTensorInfo(params.get_InputToForgetWeights()); - auto inputToCellWeights = CreateConstTensorInfo(params.get_InputToCellWeights()); - auto inputToOutputWeights = CreateConstTensorInfo(params.get_InputToOutputWeights()); - - auto recurrentToInputWeights = CreateConstTensorInfo(params.get_RecurrentToInputWeights()); - auto recurrentToForgetWeights = CreateConstTensorInfo(params.get_RecurrentToForgetWeights()); - auto recurrentToCellWeights = CreateConstTensorInfo(params.get_RecurrentToCellWeights()); - auto recurrentToOutputWeights = CreateConstTensorInfo(params.get_RecurrentToOutputWeights()); - - auto inputGateBias = CreateConstTensorInfo(params.get_InputGateBias()); - auto forgetGateBias = CreateConstTensorInfo(params.get_ForgetGateBias()); - auto cellBias = CreateConstTensorInfo(params.get_CellBias()); - auto outputGateBias = CreateConstTensorInfo(params.get_OutputGateBias()); + auto inputToInputWeights = CreateConstTensorInfo(params.GetInputToInputWeights()); + auto inputToForgetWeights = CreateConstTensorInfo(params.GetInputToForgetWeights()); + auto inputToCellWeights = CreateConstTensorInfo(params.GetInputToCellWeights()); + auto inputToOutputWeights = CreateConstTensorInfo(params.GetInputToOutputWeights()); + + auto recurrentToInputWeights = CreateConstTensorInfo(params.GetRecurrentToInputWeights()); + auto recurrentToForgetWeights = CreateConstTensorInfo(params.GetRecurrentToForgetWeights()); + auto recurrentToCellWeights = CreateConstTensorInfo(params.GetRecurrentToCellWeights()); + auto recurrentToOutputWeights = CreateConstTensorInfo(params.GetRecurrentToOutputWeights()); + + auto inputGateBias = CreateConstTensorInfo(params.GetInputGateBias()); + auto forgetGateBias = CreateConstTensorInfo(params.GetForgetGateBias()); + auto cellBias = CreateConstTensorInfo(params.GetCellBias()); + auto outputGateBias = CreateConstTensorInfo(params.GetOutputGateBias()); auto fbQuantizedLstmParams = serializer::CreateQuantizedLstmInputParams( m_flatBufferBuilder, |