diff options
Diffstat (limited to 'src/backends/cl/workloads/ClLstmFloatWorkload.hpp')
-rw-r--r-- | src/backends/cl/workloads/ClLstmFloatWorkload.hpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/backends/cl/workloads/ClLstmFloatWorkload.hpp b/src/backends/cl/workloads/ClLstmFloatWorkload.hpp index b9faca8b54..54c5c600dc 100644 --- a/src/backends/cl/workloads/ClLstmFloatWorkload.hpp +++ b/src/backends/cl/workloads/ClLstmFloatWorkload.hpp @@ -22,9 +22,14 @@ public: const WorkloadInfo& info, const arm_compute::CLCompileContext& clCompileContext); void Execute() const override; + // Replace input tensor handle with the given TensorHandle + void ReplaceInputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override; + // Replace output tensor handle with the given TensorHandle + void ReplaceOutputTensorHandle(ITensorHandle* tensorHandle, unsigned int slot) override; private: mutable arm_compute::CLLSTMLayer m_LstmLayer; + virtual void Reconfigure(); std::unique_ptr<arm_compute::CLTensor> m_InputToInputWeightsTensor; std::unique_ptr<arm_compute::CLTensor> m_InputToForgetWeightsTensor; |