From: Ferran Balaguer <ferran.balaguer@arm.com>
Date: Wed, 24 Jul 2019 16:03:45 +0000 (+0100)
Subject: IVGCVSW-3566 Fix LSTM with normalization and Cifg WorkloadFactory
X-Git-Tag: submit/tizen/20200316.035456~411
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=e30c16e7759d141c7f262988b67a7ec13758e596;p=platform%2Fupstream%2Farmnn.git

IVGCVSW-3566 Fix LSTM with normalization and Cifg WorkloadFactory

Signed-off-by: Ferran Balaguer <ferran.balaguer@arm.com>
Change-Id: I586415357d0f0d315c7174ad385167effa66b195
---

diff --git a/src/backends/backendsCommon/WorkloadFactory.cpp b/src/backends/backendsCommon/WorkloadFactory.cpp
index cbaae40..dca5778 100644
--- a/src/backends/backendsCommon/WorkloadFactory.cpp
+++ b/src/backends/backendsCommon/WorkloadFactory.cpp
@@ -460,9 +460,12 @@ bool IWorkloadFactory::IsLayerSupported(const BackendId& backendId,
 
             if(descriptor.m_LayerNormEnabled)
             {
-                optInputLayerNormWeights = OverrideDataType(
-                        cLayer->m_LayerNormParameters.m_InputLayerNormWeights->GetTensorInfo(), dataType);
-                paramsInfo.m_InputLayerNormWeights = &optInputLayerNormWeights;
+                if (!descriptor.m_CifgEnabled)
+                {
+                    optInputLayerNormWeights = OverrideDataType(
+                            cLayer->m_LayerNormParameters.m_InputLayerNormWeights->GetTensorInfo(), dataType);
+                    paramsInfo.m_InputLayerNormWeights = &optInputLayerNormWeights;
+                }
 
                 optForgetLayerNormWeights = OverrideDataType(
                         cLayer->m_LayerNormParameters.m_ForgetLayerNormWeights->GetTensorInfo(), dataType);