1 files changed, 21 insertions, 21 deletions
diff --git a/src/backends/reference/workloads/RefLstmWorkload.hpp b/src/backends/reference/workloads/RefLstmWorkload.hpp
index b55a1f9a9e..6feffbcb00 100644
--- a/src/backends/reference/workloads/RefLstmWorkload.hpp
+++ b/src/backends/reference/workloads/RefLstmWorkload.hpp
@@ -23,27 +23,27 @@ public:
 
 private:
     void Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const;
-    std::unique_ptr<ScopedCpuTensorHandle> m_InputToInputWeightsTensor;
-    std::unique_ptr<ScopedCpuTensorHandle> m_InputToForgetWeightsTensor;
-    std::unique_ptr<ScopedCpuTensorHandle> m_InputToCellWeightsTensor;
-    std::unique_ptr<ScopedCpuTensorHandle> m_InputToOutputWeightsTensor;
-    std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToInputWeightsTensor;
-    std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToForgetWeightsTensor;
-    std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToCellWeightsTensor;
-    std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToOutputWeightsTensor;
-    std::unique_ptr<ScopedCpuTensorHandle> m_CellToInputWeightsTensor;
-    std::unique_ptr<ScopedCpuTensorHandle> m_CellToForgetWeightsTensor;
-    std::unique_ptr<ScopedCpuTensorHandle> m_CellToOutputWeightsTensor;
-    std::unique_ptr<ScopedCpuTensorHandle> m_InputGateBiasTensor;
-    std::unique_ptr<ScopedCpuTensorHandle> m_ForgetGateBiasTensor;
-    std::unique_ptr<ScopedCpuTensorHandle> m_CellBiasTensor;
-    std::unique_ptr<ScopedCpuTensorHandle> m_OutputGateBiasTensor;
-    std::unique_ptr<ScopedCpuTensorHandle> m_ProjectionWeightsTensor;
-    std::unique_ptr<ScopedCpuTensorHandle> m_ProjectionBiasTensor;
-    std::unique_ptr<ScopedCpuTensorHandle> m_InputLayerNormWeights;
-    std::unique_ptr<ScopedCpuTensorHandle> m_ForgetLayerNormWeights;
-    std::unique_ptr<ScopedCpuTensorHandle> m_CellLayerNormWeights;
-    std::unique_ptr<ScopedCpuTensorHandle> m_OutputLayerNormWeights;
+    std::unique_ptr<ScopedTensorHandle> m_InputToInputWeightsTensor;
+    std::unique_ptr<ScopedTensorHandle> m_InputToForgetWeightsTensor;
+    std::unique_ptr<ScopedTensorHandle> m_InputToCellWeightsTensor;
+    std::unique_ptr<ScopedTensorHandle> m_InputToOutputWeightsTensor;
+    std::unique_ptr<ScopedTensorHandle> m_RecurrentToInputWeightsTensor;
+    std::unique_ptr<ScopedTensorHandle> m_RecurrentToForgetWeightsTensor;
+    std::unique_ptr<ScopedTensorHandle> m_RecurrentToCellWeightsTensor;
+    std::unique_ptr<ScopedTensorHandle> m_RecurrentToOutputWeightsTensor;
+    std::unique_ptr<ScopedTensorHandle> m_CellToInputWeightsTensor;
+    std::unique_ptr<ScopedTensorHandle> m_CellToForgetWeightsTensor;
+    std::unique_ptr<ScopedTensorHandle> m_CellToOutputWeightsTensor;
+    std::unique_ptr<ScopedTensorHandle> m_InputGateBiasTensor;
+    std::unique_ptr<ScopedTensorHandle> m_ForgetGateBiasTensor;
+    std::unique_ptr<ScopedTensorHandle> m_CellBiasTensor;
+    std::unique_ptr<ScopedTensorHandle> m_OutputGateBiasTensor;
+    std::unique_ptr<ScopedTensorHandle> m_ProjectionWeightsTensor;
+    std::unique_ptr<ScopedTensorHandle> m_ProjectionBiasTensor;
+    std::unique_ptr<ScopedTensorHandle> m_InputLayerNormWeights;
+    std::unique_ptr<ScopedTensorHandle> m_ForgetLayerNormWeights;
+    std::unique_ptr<ScopedTensorHandle> m_CellLayerNormWeights;
+    std::unique_ptr<ScopedTensorHandle> m_OutputLayerNormWeights;
 
     float m_LayerNormEpsilon = static_cast<float>(1e-8);
 };