aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/RefLstmFloat32Workload.hpp
diff options
context:
space:
mode:
authorNattapat Chaimanowong <nattapat.chaimanowong@arm.com>2019-05-07 12:02:30 +0100
committerderek.lamberti <derek.lamberti@arm.com>2019-05-07 12:15:38 +0000
commiteb2b329b761ce3206505ed8d2eab071a2f97d5e7 (patch)
treea236bea2077d391fdae90cdbee866f6754897623 /src/backends/reference/workloads/RefLstmFloat32Workload.hpp
parent5cf4d1c29a36bb1d675a7cbe2d24b688deb7d160 (diff)
downloadarmnn-eb2b329b761ce3206505ed8d2eab071a2f97d5e7.tar.gz
IVGCVSW-2997 Refactor reference LSTM workload
Signed-off-by: Nattapat Chaimanowong <nattapat.chaimanowong@arm.com> Change-Id: I6883f878d9f701a55153292769d2fc0530d2529e
Diffstat (limited to 'src/backends/reference/workloads/RefLstmFloat32Workload.hpp')
-rw-r--r--src/backends/reference/workloads/RefLstmFloat32Workload.hpp43
1 files changed, 0 insertions, 43 deletions
diff --git a/src/backends/reference/workloads/RefLstmFloat32Workload.hpp b/src/backends/reference/workloads/RefLstmFloat32Workload.hpp
deleted file mode 100644
index a2dead8b9c..0000000000
--- a/src/backends/reference/workloads/RefLstmFloat32Workload.hpp
+++ /dev/null
@@ -1,43 +0,0 @@
-//
-// Copyright © 2017 Arm Ltd. All rights reserved.
-// SPDX-License-Identifier: MIT
-//
-
-#pragma once
-
-#include <armnn/TypesUtils.hpp>
-
-#include <backendsCommon/Workload.hpp>
-#include <backendsCommon/WorkloadData.hpp>
-
-namespace armnn
-{
-
-class RefLstmFloat32Workload : public Float32Workload<LstmQueueDescriptor>
-{
-public:
- explicit RefLstmFloat32Workload(const LstmQueueDescriptor& descriptor, const WorkloadInfo& info);
-
- virtual void Execute() const override;
-
-private:
- 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;
-};
-
-} //namespace armnn