From 6940dd720ebb6b3d1df8ca203ab696daefe58189 Mon Sep 17 00:00:00 2001 From: Jim Flynn Date: Fri, 20 Mar 2020 12:25:56 +0000 Subject: renamed Documentation folder 20.02 and added .nojekyll file Signed-off-by: Jim Flynn --- .../_cl_quantized_lstm_workload_8hpp_source.xhtml | 126 +++++++++++++++++++++ 1 file changed, 126 insertions(+) create mode 100644 20.02/_cl_quantized_lstm_workload_8hpp_source.xhtml (limited to '20.02/_cl_quantized_lstm_workload_8hpp_source.xhtml') diff --git a/20.02/_cl_quantized_lstm_workload_8hpp_source.xhtml b/20.02/_cl_quantized_lstm_workload_8hpp_source.xhtml new file mode 100644 index 0000000000..5195051450 --- /dev/null +++ b/20.02/_cl_quantized_lstm_workload_8hpp_source.xhtml @@ -0,0 +1,126 @@ + + + + + + + + + + + + + +ArmNN: src/backends/cl/workloads/ClQuantizedLstmWorkload.hpp Source File + + + + + + + + + + + + + + + + +
+
+ + + + ArmNN + + + +
+
+  20.02 +
+
+
+ + + + + + + +
+
+ +
+
+
+ +
+ +
+
+ + +
+ +
+ +
+
+
ClQuantizedLstmWorkload.hpp
+
+
+Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include <armnn/QuantizedLstmParams.hpp>
9 #include <backendsCommon/Workload.hpp>
10 #include <backendsCommon/WorkloadData.hpp>
11 
12 #include <arm_compute/runtime/CL/functions/CLLSTMLayerQuantized.h>
13 
14 namespace armnn
15 {
16 
17 arm_compute::Status ClQuantizedLstmWorkloadValidate(const TensorInfo& input, const TensorInfo& previousCellStateIn,
18  const TensorInfo& previousOutputIn, const TensorInfo& cellStateOut,
19  const TensorInfo& output,
20  const QuantizedLstmInputParamsInfo& paramsInfo);
21 
22 class ClQuantizedLstmWorkload : public BaseWorkload<QuantizedLstmQueueDescriptor>
23 {
24 public:
25  ClQuantizedLstmWorkload(const QuantizedLstmQueueDescriptor& descriptor, const WorkloadInfo& info);
26  void Execute() const override;
27 
28 private:
29  mutable arm_compute::CLLSTMLayerQuantized m_QuantizedLstmLayer;
30 
31  std::unique_ptr<arm_compute::CLTensor> m_InputToInputWeightsTensor;
32  std::unique_ptr<arm_compute::CLTensor> m_InputToForgetWeightsTensor;
33  std::unique_ptr<arm_compute::CLTensor> m_InputToCellWeightsTensor;
34  std::unique_ptr<arm_compute::CLTensor> m_InputToOutputWeightsTensor;
35  std::unique_ptr<arm_compute::CLTensor> m_RecurrentToInputWeightsTensor;
36  std::unique_ptr<arm_compute::CLTensor> m_RecurrentToForgetWeightsTensor;
37  std::unique_ptr<arm_compute::CLTensor> m_RecurrentToCellWeightsTensor;
38  std::unique_ptr<arm_compute::CLTensor> m_RecurrentToOutputWeightsTensor;
39  std::unique_ptr<arm_compute::CLTensor> m_InputGateBiasTensor;
40  std::unique_ptr<arm_compute::CLTensor> m_ForgetGateBiasTensor;
41  std::unique_ptr<arm_compute::CLTensor> m_CellBiasTensor;
42  std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor;
43 
44  void FreeUnusedTensors();
45 };
46 
47 } //namespace armnn
48 
49 
WorkloadData.hpp
+ + +
QuantizedLstmParams.hpp
+
armnn
Copyright (c) 2020 ARM Limited.
Definition: 00_introduction.dox:25
+ +
armnn::ClQuantizedLstmWorkload::ClQuantizedLstmWorkload
ClQuantizedLstmWorkload(const QuantizedLstmQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition: ClQuantizedLstmWorkload.cpp:64
+
armnn::Status
Status
enumeration
Definition: Types.hpp:26
+
Workload.hpp
+ + +
armnn::WorkloadInfo
Contains information about inputs and outputs to a layer.
Definition: WorkloadInfo.hpp:16
+
armnn::ClQuantizedLstmWorkloadValidate
arm_compute::Status ClQuantizedLstmWorkloadValidate(const TensorInfo &input, const TensorInfo &previousCellStateIn, const TensorInfo &previousOutputIn, const TensorInfo &cellStateOut, const TensorInfo &output, const QuantizedLstmInputParamsInfo &paramsInfo)
Definition: ClQuantizedLstmWorkload.cpp:18
+
+
+ + + + -- cgit v1.2.1