diff options
Diffstat (limited to 'src/backends/reference/RefLayerSupport.cpp')
| -rw-r--r-- | src/backends/reference/RefLayerSupport.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp index 2603371927..5eba3e5919 100644 --- a/src/backends/reference/RefLayerSupport.cpp +++ b/src/backends/reference/RefLayerSupport.cpp @@ -2317,9 +2317,10 @@ bool RefLayerSupport::IsUnidirectionalSequenceLstmSupported( DataType::Float32 }; - std::array<DataType, 1> supportedWeightTypes = + std::array<DataType, 2> supportedWeightTypes = { - DataType::Float32 + DataType::Float32, + DataType::QAsymmS8 }; // check inputs and outputs |