| /aosp_15_r20/external/armnn/src/backends/neon/workloads/ |
| H A D | NeonQLstmWorkload.cpp | 67 m_CellToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload()
68 …BuildArmComputeTensor(*m_CellToForgetWeightsTensor, m_Data.m_CellToForgetWeights->GetTensorInfo()); in NeonQLstmWorkload()
74 qLstmParams.set_peephole_params(m_CellToForgetWeightsTensor.get(), in NeonQLstmWorkload()
215 InitializeArmComputeTensorData(*m_CellToForgetWeightsTensor, m_Data.m_CellToForgetWeights); in NeonQLstmWorkload()
406 FreeTensorIfUnused(m_CellToForgetWeightsTensor); in FreeUnusedTensors()
|
| H A D | NeonLstmFloatWorkload.cpp | 100 m_CellToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload()
101 …BuildArmComputeTensor(*m_CellToForgetWeightsTensor, m_Data.m_CellToForgetWeights->GetTensorInfo()); in NeonLstmFloatWorkload()
106 …lstm_param.set_peephole_params(m_CellToForgetWeightsTensor.get(), m_CellToOutputWeightsTensor.get(… in NeonLstmFloatWorkload()
224 InitializeArmComputeTensorData(*m_CellToForgetWeightsTensor, in NeonLstmFloatWorkload()
402 FreeTensorIfUnused(m_CellToForgetWeightsTensor); in FreeUnusedTensors()
|
| H A D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 219 m_CellToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmFloatWorkload()
220 …BuildArmComputeTensor(*m_CellToForgetWeightsTensor, m_Data.m_CellToForgetWeights->GetTensorInfo()); in NeonUnidirectionalSequenceLstmFloatWorkload()
225 …lstm_param.set_peephole_params(m_CellToForgetWeightsTensor.get(), m_CellToOutputWeightsTensor.get(… in NeonUnidirectionalSequenceLstmFloatWorkload()
386 InitializeArmComputeTensorData(*m_CellToForgetWeightsTensor, m_Data.m_CellToForgetWeights); in NeonUnidirectionalSequenceLstmFloatWorkload()
892 FreeTensorIfUnused(m_CellToForgetWeightsTensor); in FreeUnusedTensors()
|
| H A D | NeonUnidirectionalSequenceLstmWorkload.cpp | 231 m_CellToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmWorkload()
232 …BuildArmComputeTensor(*m_CellToForgetWeightsTensor, m_Data.m_CellToForgetWeights->GetTensorInfo()); in NeonUnidirectionalSequenceLstmWorkload()
237 …lstm_param.set_peephole_params(m_CellToForgetWeightsTensor.get(), m_CellToOutputWeightsTensor.get(… in NeonUnidirectionalSequenceLstmWorkload()
367 InitializeArmComputeTensorData(*m_CellToForgetWeightsTensor, m_Data.m_CellToForgetWeights); in NeonUnidirectionalSequenceLstmWorkload()
865 FreeTensorIfUnused(m_CellToForgetWeightsTensor); in FreeUnusedTensors()
|
| H A D | NeonQLstmWorkload.hpp | 39 std::unique_ptr<arm_compute::Tensor> m_CellToForgetWeightsTensor; member in armnn::NeonQLstmWorkload
|
| H A D | NeonLstmFloatWorkload.hpp | 41 std::unique_ptr<arm_compute::Tensor> m_CellToForgetWeightsTensor; member in armnn::NeonLstmFloatWorkload
|
| H A D | NeonUnidirectionalSequenceLstmWorkload.hpp | 52 std::unique_ptr<arm_compute::Tensor> m_CellToForgetWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmWorkload
|
| H A D | NeonUnidirectionalSequenceLstmFloatWorkload.hpp | 54 std::unique_ptr<arm_compute::Tensor> m_CellToForgetWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmFloatWorkload
|
| /aosp_15_r20/external/armnn/src/backends/cl/workloads/ |
| H A D | ClQLstmWorkload.cpp | 69 m_CellToForgetWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload()
70 …BuildArmComputeTensor(*m_CellToForgetWeightsTensor, m_Data.m_CellToForgetWeights->GetTensorInfo()); in ClQLstmWorkload()
76 qLstmParams.set_peephole_params(m_CellToForgetWeightsTensor.get(), in ClQLstmWorkload()
221 … InitializeArmComputeClTensorData(*m_CellToForgetWeightsTensor, m_Data.m_CellToForgetWeights); in ClQLstmWorkload()
410 FreeTensorIfUnused(m_CellToForgetWeightsTensor); in FreeUnusedTensors()
|
| H A D | ClLstmFloatWorkload.cpp | 105 m_CellToForgetWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload()
106 …BuildArmComputeTensor(*m_CellToForgetWeightsTensor, m_Data.m_CellToForgetWeights->GetTensorInfo()); in ClLstmFloatWorkload()
111 …lstm_param.set_peephole_params(m_CellToForgetWeightsTensor.get(), m_CellToOutputWeightsTensor.get(… in ClLstmFloatWorkload()
216 … InitializeArmComputeClTensorData(*m_CellToForgetWeightsTensor, m_Data.m_CellToForgetWeights); in ClLstmFloatWorkload()
381 FreeTensorIfUnused(m_CellToForgetWeightsTensor); in FreeUnusedTensors()
|
| H A D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 221 m_CellToForgetWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClUnidirectionalSequenceLstmFloatWorkload()
222 …BuildArmComputeTensor(*m_CellToForgetWeightsTensor, m_Data.m_CellToForgetWeights->GetTensorInfo()); in ClUnidirectionalSequenceLstmFloatWorkload()
227 …lstm_param.set_peephole_params(m_CellToForgetWeightsTensor.get(), m_CellToOutputWeightsTensor.get(… in ClUnidirectionalSequenceLstmFloatWorkload()
383 … InitializeArmComputeClTensorData(*m_CellToForgetWeightsTensor, m_Data.m_CellToForgetWeights); in ClUnidirectionalSequenceLstmFloatWorkload()
888 FreeTensorIfUnused(m_CellToForgetWeightsTensor); in FreeUnusedTensors()
|
| H A D | ClQLstmWorkload.hpp | 41 std::unique_ptr<arm_compute::CLTensor> m_CellToForgetWeightsTensor; member in armnn::ClQLstmWorkload
|
| H A D | ClLstmFloatWorkload.hpp | 43 std::unique_ptr<arm_compute::CLTensor> m_CellToForgetWeightsTensor; member in armnn::ClLstmFloatWorkload
|
| H A D | ClUnidirectionalSequenceLstmFloatWorkload.hpp | 56 std::unique_ptr<arm_compute::CLTensor> m_CellToForgetWeightsTensor; member in armnn::ClUnidirectionalSequenceLstmFloatWorkload
|
| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | RefLstmWorkload.cpp | 28 , m_CellToForgetWeightsTensor (AssignScopedTensorHandle(descriptor.m_CellToForgetWeights)) in RefLstmWorkload()
186 …m_CellToForgetWeightsTensor->GetTensorInfo(), m_CellToForgetWeightsTensor->GetConstTensor<void>()); in Execute()
|
| H A D | RefUnidirectionalSequenceLstmWorkload.cpp | 32 , m_CellToForgetWeightsTensor (AssignScopedTensorHandle(descriptor.m_CellToForgetWeights)) in RefUnidirectionalSequenceLstmWorkload()
214 …m_CellToForgetWeightsTensor->GetTensorInfo(), m_CellToForgetWeightsTensor->GetConstTensor<void>()); in Execute()
|
| H A D | RefQLstmWorkload.cpp | 29 … , m_CellToForgetWeightsTensor (AssignScopedTensorHandle(descriptor.m_CellToForgetWeights)) in RefQLstmWorkload()
228 …m_CellToForgetWeightsTensor->GetTensorInfo(), m_CellToForgetWeightsTensor->GetConstTensor<void>()); in Execute()
|
| H A D | RefUnidirectionalSequenceLstmWorkload.hpp | 40 std::unique_ptr<ScopedTensorHandle> m_CellToForgetWeightsTensor; member in armnn::RefUnidirectionalSequenceLstmWorkload
|
| H A D | RefLstmWorkload.hpp | 35 std::unique_ptr<ScopedTensorHandle> m_CellToForgetWeightsTensor; member in armnn::RefLstmWorkload
|
| H A D | RefQLstmWorkload.hpp | 37 std::unique_ptr<ScopedTensorHandle> m_CellToForgetWeightsTensor; member in armnn::RefQLstmWorkload
|