diff options
Diffstat (limited to 'src/backends/reference/RefLayerSupport.cpp')
| -rw-r--r-- | src/backends/reference/RefLayerSupport.cpp | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp index b563badca5..3d260c5abd 100644 --- a/src/backends/reference/RefLayerSupport.cpp +++ b/src/backends/reference/RefLayerSupport.cpp @@ -861,7 +861,11 @@ bool RefLayerSupport::IsLstmSupported(const TensorInfo& input, const TensorInfo* projectionBias, const TensorInfo* cellToForgetWeights, const TensorInfo* cellToOutputWeights, - Optional<std::string&> reasonIfUnsupported) const + Optional<std::string&> reasonIfUnsupported, + const TensorInfo* inputLayerNormWeights, + const TensorInfo* forgetLayerNormWeights, + const TensorInfo* cellLayerNormWeights, + const TensorInfo* outputLayerNormWeights) const { ignore_unused(descriptor); ignore_unused(inputToForgetWeights); @@ -881,6 +885,10 @@ bool RefLayerSupport::IsLstmSupported(const TensorInfo& input, ignore_unused(projectionBias); ignore_unused(cellToForgetWeights); ignore_unused(cellToOutputWeights); + ignore_unused(inputLayerNormWeights); + ignore_unused(forgetLayerNormWeights); + ignore_unused(cellLayerNormWeights); + ignore_unused(outputLayerNormWeights); bool supported = true; |