| /aosp_15_r20/external/armnn/src/backends/neon/workloads/ |
| H A D | NeonQLstmWorkload.cpp | 70 m_CellToOutputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload()
71 …BuildArmComputeTensor(*m_CellToOutputWeightsTensor, m_Data.m_CellToOutputWeights->GetTensorInfo()); in NeonQLstmWorkload()
75 m_CellToOutputWeightsTensor.get()); in NeonQLstmWorkload()
216 InitializeArmComputeTensorData(*m_CellToOutputWeightsTensor, m_Data.m_CellToOutputWeights); in NeonQLstmWorkload()
407 FreeTensorIfUnused(m_CellToOutputWeightsTensor); in FreeUnusedTensors()
|
| H A D | NeonLstmFloatWorkload.cpp | 103 m_CellToOutputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload()
104 …BuildArmComputeTensor(*m_CellToOutputWeightsTensor, m_Data.m_CellToOutputWeights->GetTensorInfo()); in NeonLstmFloatWorkload()
106 …tm_param.set_peephole_params(m_CellToForgetWeightsTensor.get(), m_CellToOutputWeightsTensor.get()); in NeonLstmFloatWorkload()
226 InitializeArmComputeTensorData(*m_CellToOutputWeightsTensor, in NeonLstmFloatWorkload()
403 FreeTensorIfUnused(m_CellToOutputWeightsTensor); in FreeUnusedTensors()
|
| H A D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 222 m_CellToOutputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmFloatWorkload()
223 …BuildArmComputeTensor(*m_CellToOutputWeightsTensor, m_Data.m_CellToOutputWeights->GetTensorInfo()); in NeonUnidirectionalSequenceLstmFloatWorkload()
225 …tm_param.set_peephole_params(m_CellToForgetWeightsTensor.get(), m_CellToOutputWeightsTensor.get()); in NeonUnidirectionalSequenceLstmFloatWorkload()
387 InitializeArmComputeTensorData(*m_CellToOutputWeightsTensor, m_Data.m_CellToOutputWeights); in NeonUnidirectionalSequenceLstmFloatWorkload()
893 FreeTensorIfUnused(m_CellToOutputWeightsTensor); in FreeUnusedTensors()
|
| H A D | NeonUnidirectionalSequenceLstmWorkload.cpp | 234 m_CellToOutputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmWorkload()
235 …BuildArmComputeTensor(*m_CellToOutputWeightsTensor, m_Data.m_CellToOutputWeights->GetTensorInfo()); in NeonUnidirectionalSequenceLstmWorkload()
237 …tm_param.set_peephole_params(m_CellToForgetWeightsTensor.get(), m_CellToOutputWeightsTensor.get()); in NeonUnidirectionalSequenceLstmWorkload()
368 InitializeArmComputeTensorData(*m_CellToOutputWeightsTensor, m_Data.m_CellToOutputWeights); in NeonUnidirectionalSequenceLstmWorkload()
866 FreeTensorIfUnused(m_CellToOutputWeightsTensor); in FreeUnusedTensors()
|
| H A D | NeonQLstmWorkload.hpp | 40 std::unique_ptr<arm_compute::Tensor> m_CellToOutputWeightsTensor; member in armnn::NeonQLstmWorkload
|
| H A D | NeonLstmFloatWorkload.hpp | 42 std::unique_ptr<arm_compute::Tensor> m_CellToOutputWeightsTensor; member in armnn::NeonLstmFloatWorkload
|
| H A D | NeonUnidirectionalSequenceLstmWorkload.hpp | 53 std::unique_ptr<arm_compute::Tensor> m_CellToOutputWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmWorkload
|
| H A D | NeonUnidirectionalSequenceLstmFloatWorkload.hpp | 55 std::unique_ptr<arm_compute::Tensor> m_CellToOutputWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmFloatWorkload
|
| /aosp_15_r20/external/armnn/src/backends/cl/workloads/ |
| H A D | ClQLstmWorkload.cpp | 72 m_CellToOutputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload()
73 …BuildArmComputeTensor(*m_CellToOutputWeightsTensor, m_Data.m_CellToOutputWeights->GetTensorInfo()); in ClQLstmWorkload()
77 m_CellToOutputWeightsTensor.get()); in ClQLstmWorkload()
222 … InitializeArmComputeClTensorData(*m_CellToOutputWeightsTensor, m_Data.m_CellToOutputWeights); in ClQLstmWorkload()
411 FreeTensorIfUnused(m_CellToOutputWeightsTensor); in FreeUnusedTensors()
|
| H A D | ClLstmFloatWorkload.cpp | 108 m_CellToOutputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload()
109 …BuildArmComputeTensor(*m_CellToOutputWeightsTensor, m_Data.m_CellToOutputWeights->GetTensorInfo()); in ClLstmFloatWorkload()
111 …tm_param.set_peephole_params(m_CellToForgetWeightsTensor.get(), m_CellToOutputWeightsTensor.get()); in ClLstmFloatWorkload()
217 … InitializeArmComputeClTensorData(*m_CellToOutputWeightsTensor, m_Data.m_CellToOutputWeights); in ClLstmFloatWorkload()
382 FreeTensorIfUnused(m_CellToOutputWeightsTensor); in FreeUnusedTensors()
|
| H A D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 224 m_CellToOutputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClUnidirectionalSequenceLstmFloatWorkload()
225 …BuildArmComputeTensor(*m_CellToOutputWeightsTensor, m_Data.m_CellToOutputWeights->GetTensorInfo()); in ClUnidirectionalSequenceLstmFloatWorkload()
227 …tm_param.set_peephole_params(m_CellToForgetWeightsTensor.get(), m_CellToOutputWeightsTensor.get()); in ClUnidirectionalSequenceLstmFloatWorkload()
384 … InitializeArmComputeClTensorData(*m_CellToOutputWeightsTensor, m_Data.m_CellToOutputWeights); in ClUnidirectionalSequenceLstmFloatWorkload()
889 FreeTensorIfUnused(m_CellToOutputWeightsTensor); in FreeUnusedTensors()
|
| H A D | ClQLstmWorkload.hpp | 42 std::unique_ptr<arm_compute::CLTensor> m_CellToOutputWeightsTensor; member in armnn::ClQLstmWorkload
|
| H A D | ClLstmFloatWorkload.hpp | 44 std::unique_ptr<arm_compute::CLTensor> m_CellToOutputWeightsTensor; member in armnn::ClLstmFloatWorkload
|
| H A D | ClUnidirectionalSequenceLstmFloatWorkload.hpp | 57 std::unique_ptr<arm_compute::CLTensor> m_CellToOutputWeightsTensor; member in armnn::ClUnidirectionalSequenceLstmFloatWorkload
|
| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | RefLstmWorkload.cpp | 29 , m_CellToOutputWeightsTensor (AssignScopedTensorHandle(descriptor.m_CellToOutputWeights)) in RefLstmWorkload()
188 …m_CellToOutputWeightsTensor->GetTensorInfo(), m_CellToOutputWeightsTensor->GetConstTensor<void>()); in Execute()
|
| H A D | RefUnidirectionalSequenceLstmWorkload.cpp | 33 , m_CellToOutputWeightsTensor (AssignScopedTensorHandle(descriptor.m_CellToOutputWeights)) in RefUnidirectionalSequenceLstmWorkload()
216 …m_CellToOutputWeightsTensor->GetTensorInfo(), m_CellToOutputWeightsTensor->GetConstTensor<void>()); in Execute()
|
| H A D | RefQLstmWorkload.cpp | 30 … , m_CellToOutputWeightsTensor (AssignScopedTensorHandle(descriptor.m_CellToOutputWeights)) in RefQLstmWorkload()
230 …m_CellToOutputWeightsTensor->GetTensorInfo(), m_CellToOutputWeightsTensor->GetConstTensor<void>()); in Execute()
|
| H A D | RefUnidirectionalSequenceLstmWorkload.hpp | 41 std::unique_ptr<ScopedTensorHandle> m_CellToOutputWeightsTensor; member in armnn::RefUnidirectionalSequenceLstmWorkload
|
| H A D | RefLstmWorkload.hpp | 36 std::unique_ptr<ScopedTensorHandle> m_CellToOutputWeightsTensor; member in armnn::RefLstmWorkload
|
| H A D | RefQLstmWorkload.hpp | 38 std::unique_ptr<ScopedTensorHandle> m_CellToOutputWeightsTensor; member in armnn::RefQLstmWorkload
|