/aosp_15_r20/external/armnn/src/backends/neon/workloads/ |
H A D | NeonQuantizedLstmWorkload.cpp | 37 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQuantizedLstmWorkload() 38 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in NeonQuantizedLstmWorkload() 71 m_RecurrentToForgetWeightsTensor.get(), in NeonQuantizedLstmWorkload() 98 InitializeArmComputeTensorData(*m_RecurrentToForgetWeightsTensor, in NeonQuantizedLstmWorkload() 199 FreeTensorIfUnused(m_RecurrentToForgetWeightsTensor); in FreeUnusedTensors()
|
H A D | NeonQLstmWorkload.cpp | 38 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload() 39 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in NeonQLstmWorkload() 164 m_RecurrentToForgetWeightsTensor.get(), in NeonQLstmWorkload() 182 …InitializeArmComputeTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeight… in NeonQLstmWorkload() 401 FreeTensorIfUnused(m_RecurrentToForgetWeightsTensor); in FreeUnusedTensors()
|
H A D | NeonLstmFloatWorkload.cpp | 41 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload() 42 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in NeonLstmFloatWorkload() 168 … m_InputToOutputWeightsTensor.get(), m_RecurrentToForgetWeightsTensor.get(), in NeonLstmFloatWorkload() 183 InitializeArmComputeTensorData(*m_RecurrentToForgetWeightsTensor, in NeonLstmFloatWorkload() 398 FreeTensorIfUnused(m_RecurrentToForgetWeightsTensor); in FreeUnusedTensors()
|
H A D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 160 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmFloatWorkload() 161 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in NeonUnidirectionalSequenceLstmFloatWorkload() 332 m_RecurrentToForgetWeightsTensor.get(), in NeonUnidirectionalSequenceLstmFloatWorkload() 357 …InitializeArmComputeTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeight… in NeonUnidirectionalSequenceLstmFloatWorkload() 888 FreeTensorIfUnused(m_RecurrentToForgetWeightsTensor); in FreeUnusedTensors()
|
H A D | NeonUnidirectionalSequenceLstmWorkload.cpp | 173 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmWorkload() 174 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in NeonUnidirectionalSequenceLstmWorkload() 319 m_RecurrentToForgetWeightsTensor.get(), in NeonUnidirectionalSequenceLstmWorkload() 338 …InitializeArmComputeTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeight… in NeonUnidirectionalSequenceLstmWorkload() 861 FreeTensorIfUnused(m_RecurrentToForgetWeightsTensor); in FreeUnusedTensors()
|
H A D | NeonQuantizedLstmWorkload.hpp | 33 std::unique_ptr<arm_compute::Tensor> m_RecurrentToForgetWeightsTensor; member in armnn::NeonQuantizedLstmWorkload
|
H A D | NeonQLstmWorkload.hpp | 34 std::unique_ptr<arm_compute::Tensor> m_RecurrentToForgetWeightsTensor; member in armnn::NeonQLstmWorkload
|
H A D | NeonLstmFloatWorkload.hpp | 37 std::unique_ptr<arm_compute::Tensor> m_RecurrentToForgetWeightsTensor; member in armnn::NeonLstmFloatWorkload
|
H A D | NeonUnidirectionalSequenceLstmWorkload.hpp | 48 std::unique_ptr<arm_compute::Tensor> m_RecurrentToForgetWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmWorkload
|
H A D | NeonUnidirectionalSequenceLstmFloatWorkload.hpp | 50 std::unique_ptr<arm_compute::Tensor> m_RecurrentToForgetWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmFloatWorkload
|
/aosp_15_r20/external/armnn/src/backends/cl/workloads/ |
H A D | ClQuantizedLstmWorkload.cpp | 84 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQuantizedLstmWorkload() 85 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in ClQuantizedLstmWorkload() 117 … m_RecurrentToInputWeightsTensor.get(), m_RecurrentToForgetWeightsTensor.get(), in ClQuantizedLstmWorkload() 130 …InitializeArmComputeClTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeig… in ClQuantizedLstmWorkload() 155 FreeTensorIfUnused(m_RecurrentToForgetWeightsTensor); in FreeUnusedTensors()
|
H A D | ClQLstmWorkload.cpp | 40 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload() 41 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in ClQLstmWorkload() 169 m_RecurrentToForgetWeightsTensor.get(), in ClQLstmWorkload() 188 …InitializeArmComputeClTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeig… in ClQLstmWorkload() 405 FreeTensorIfUnused(m_RecurrentToForgetWeightsTensor); in FreeUnusedTensors()
|
H A D | ClLstmFloatWorkload.cpp | 46 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload() 47 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in ClLstmFloatWorkload() 174 … m_RecurrentToForgetWeightsTensor.get(), m_RecurrentToCellWeightsTensor.get(), in ClLstmFloatWorkload() 187 …InitializeArmComputeClTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeig… in ClLstmFloatWorkload() 377 FreeTensorIfUnused(m_RecurrentToForgetWeightsTensor); in FreeUnusedTensors()
|
H A D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 162 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClUnidirectionalSequenceLstmFloatWorkload() 163 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in ClUnidirectionalSequenceLstmFloatWorkload() 329 m_RecurrentToForgetWeightsTensor.get(), in ClUnidirectionalSequenceLstmFloatWorkload() 354 …InitializeArmComputeClTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeig… in ClUnidirectionalSequenceLstmFloatWorkload() 884 FreeTensorIfUnused(m_RecurrentToForgetWeightsTensor); in FreeUnusedTensors()
|
H A D | ClQuantizedLstmWorkload.hpp | 40 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToForgetWeightsTensor; member in armnn::ClQuantizedLstmWorkload
|
H A D | ClQLstmWorkload.hpp | 36 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToForgetWeightsTensor; member in armnn::ClQLstmWorkload
|
H A D | ClLstmFloatWorkload.hpp | 39 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToForgetWeightsTensor; member in armnn::ClLstmFloatWorkload
|
H A D | ClUnidirectionalSequenceLstmFloatWorkload.hpp | 52 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToForgetWeightsTensor; member in armnn::ClUnidirectionalSequenceLstmFloatWorkload
|
/aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
H A D | RefLstmWorkload.cpp | 24 …, m_RecurrentToForgetWeightsTensor(AssignScopedTensorHandle(descriptor.m_RecurrentToForgetWeights)) in RefLstmWorkload() 129 …m_RecurrentToForgetWeightsTensor->GetTensorInfo(), m_RecurrentToForgetWeightsTensor->GetConstTenso… in Execute()
|
H A D | RefUnidirectionalSequenceLstmWorkload.cpp | 28 …, m_RecurrentToForgetWeightsTensor(AssignScopedTensorHandle(descriptor.m_RecurrentToForgetWeights)) in RefUnidirectionalSequenceLstmWorkload() 160 …m_RecurrentToForgetWeightsTensor->GetTensorInfo(), m_RecurrentToForgetWeightsTensor->GetConstTenso… in Execute()
|
H A D | RefQLstmWorkload.cpp | 24 …, m_RecurrentToForgetWeightsTensor(AssignScopedTensorHandle(descriptor.m_RecurrentToForgetWeights)) in RefQLstmWorkload() 123 m_RecurrentToForgetWeightsTensor->GetTensorInfo(), in Execute() 124 m_RecurrentToForgetWeightsTensor->GetConstTensor<void>()); in Execute()
|
H A D | RefUnidirectionalSequenceLstmWorkload.hpp | 36 std::unique_ptr<ScopedTensorHandle> m_RecurrentToForgetWeightsTensor; member in armnn::RefUnidirectionalSequenceLstmWorkload
|
H A D | RefLstmWorkload.hpp | 31 std::unique_ptr<ScopedTensorHandle> m_RecurrentToForgetWeightsTensor; member in armnn::RefLstmWorkload
|
H A D | RefQLstmWorkload.hpp | 32 std::unique_ptr<ScopedTensorHandle> m_RecurrentToForgetWeightsTensor; member in armnn::RefQLstmWorkload
|