| /aosp_15_r20/external/armnn/src/backends/neon/workloads/ |
| H A D | NeonQLstmWorkload.cpp | 59 m_CellToInputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload()
64 … BuildArmComputeTensor(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights->GetTensorInfo()); in NeonQLstmWorkload()
136 m_Data.m_CellToInputWeights != nullptr ? m_CellToInputWeightsTensor.get() : nullptr, in NeonQLstmWorkload()
212 … InitializeArmComputeTensorData(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights); in NeonQLstmWorkload()
405 FreeTensorIfUnused(m_CellToInputWeightsTensor); in FreeUnusedTensors()
|
| H A D | NeonLstmFloatWorkload.cpp | 68 m_CellToInputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload()
71 … BuildArmComputeTensor(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights->GetTensorInfo()); in NeonLstmFloatWorkload()
79 … m_Data.m_CellToInputWeights != nullptr ? m_CellToInputWeightsTensor.get() : nullptr, in NeonLstmFloatWorkload()
204 InitializeArmComputeTensorData(*m_CellToInputWeightsTensor, in NeonLstmFloatWorkload()
401 FreeTensorIfUnused(m_CellToInputWeightsTensor); in FreeUnusedTensors()
|
| H A D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 187 m_CellToInputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmFloatWorkload()
190 … BuildArmComputeTensor(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights->GetTensorInfo()); in NeonUnidirectionalSequenceLstmFloatWorkload()
198 … m_Data.m_CellToInputWeights ? m_CellToInputWeightsTensor.get() : nullptr, in NeonUnidirectionalSequenceLstmFloatWorkload()
370 … InitializeArmComputeTensorData(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights); in NeonUnidirectionalSequenceLstmFloatWorkload()
891 FreeTensorIfUnused(m_CellToInputWeightsTensor); in FreeUnusedTensors()
|
| H A D | NeonUnidirectionalSequenceLstmWorkload.cpp | 200 m_CellToInputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmWorkload()
203 … BuildArmComputeTensor(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights->GetTensorInfo()); in NeonUnidirectionalSequenceLstmWorkload()
210 … m_Data.m_CellToInputWeights ? m_CellToInputWeightsTensor.get() : nullptr, in NeonUnidirectionalSequenceLstmWorkload()
351 … InitializeArmComputeTensorData(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights); in NeonUnidirectionalSequenceLstmWorkload()
864 FreeTensorIfUnused(m_CellToInputWeightsTensor); in FreeUnusedTensors()
|
| H A D | NeonQLstmWorkload.hpp | 38 std::unique_ptr<arm_compute::Tensor> m_CellToInputWeightsTensor; member in armnn::NeonQLstmWorkload
|
| H A D | NeonLstmFloatWorkload.hpp | 40 std::unique_ptr<arm_compute::Tensor> m_CellToInputWeightsTensor; member in armnn::NeonLstmFloatWorkload
|
| H A D | NeonUnidirectionalSequenceLstmWorkload.hpp | 51 std::unique_ptr<arm_compute::Tensor> m_CellToInputWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmWorkload
|
| H A D | NeonUnidirectionalSequenceLstmFloatWorkload.hpp | 53 std::unique_ptr<arm_compute::Tensor> m_CellToInputWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmFloatWorkload
|
| /aosp_15_r20/external/armnn/src/backends/cl/workloads/ |
| H A D | ClQLstmWorkload.cpp | 61 m_CellToInputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload()
66 … BuildArmComputeTensor(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights->GetTensorInfo()); in ClQLstmWorkload()
138 m_Data.m_CellToInputWeights != nullptr ? m_CellToInputWeightsTensor.get() : nullptr, in ClQLstmWorkload()
218 … InitializeArmComputeClTensorData(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights); in ClQLstmWorkload()
409 FreeTensorIfUnused(m_CellToInputWeightsTensor); in FreeUnusedTensors()
|
| H A D | ClLstmFloatWorkload.cpp | 73 m_CellToInputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload()
76 … BuildArmComputeTensor(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights->GetTensorInfo()); in ClLstmFloatWorkload()
84 … m_Data.m_CellToInputWeights != nullptr ? m_CellToInputWeightsTensor.get() : nullptr, in ClLstmFloatWorkload()
200 … InitializeArmComputeClTensorData(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights); in ClLstmFloatWorkload()
380 FreeTensorIfUnused(m_CellToInputWeightsTensor); in FreeUnusedTensors()
|
| H A D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 189 m_CellToInputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClUnidirectionalSequenceLstmFloatWorkload()
192 … BuildArmComputeTensor(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights->GetTensorInfo()); in ClUnidirectionalSequenceLstmFloatWorkload()
200 … m_Data.m_CellToInputWeights ? m_CellToInputWeightsTensor.get() : nullptr, in ClUnidirectionalSequenceLstmFloatWorkload()
367 … InitializeArmComputeClTensorData(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights); in ClUnidirectionalSequenceLstmFloatWorkload()
887 FreeTensorIfUnused(m_CellToInputWeightsTensor); in FreeUnusedTensors()
|
| H A D | ClQLstmWorkload.hpp | 40 std::unique_ptr<arm_compute::CLTensor> m_CellToInputWeightsTensor; member in armnn::ClQLstmWorkload
|
| H A D | ClLstmFloatWorkload.hpp | 42 std::unique_ptr<arm_compute::CLTensor> m_CellToInputWeightsTensor; member in armnn::ClLstmFloatWorkload
|
| H A D | ClUnidirectionalSequenceLstmFloatWorkload.hpp | 55 std::unique_ptr<arm_compute::CLTensor> m_CellToInputWeightsTensor; member in armnn::ClUnidirectionalSequenceLstmFloatWorkload
|
| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | RefLstmWorkload.cpp | 27 , m_CellToInputWeightsTensor (AssignScopedTensorHandle(descriptor.m_CellToInputWeights)) in RefLstmWorkload()
194 … m_CellToInputWeightsTensor->GetTensorInfo(), m_CellToInputWeightsTensor->GetConstTensor<void>()); in Execute()
|
| H A D | RefUnidirectionalSequenceLstmWorkload.cpp | 31 , m_CellToInputWeightsTensor (AssignScopedTensorHandle(descriptor.m_CellToInputWeights)) in RefUnidirectionalSequenceLstmWorkload()
222 … m_CellToInputWeightsTensor->GetTensorInfo(), m_CellToInputWeightsTensor->GetConstTensor<void>()); in Execute()
|
| H A D | RefQLstmWorkload.cpp | 28 … , m_CellToInputWeightsTensor (AssignScopedTensorHandle(descriptor.m_CellToInputWeights)) in RefQLstmWorkload()
225 … m_CellToInputWeightsTensor->GetTensorInfo(), m_CellToInputWeightsTensor->GetConstTensor<void>()); in Execute()
|
| H A D | RefUnidirectionalSequenceLstmWorkload.hpp | 39 std::unique_ptr<ScopedTensorHandle> m_CellToInputWeightsTensor; member in armnn::RefUnidirectionalSequenceLstmWorkload
|
| H A D | RefLstmWorkload.hpp | 34 std::unique_ptr<ScopedTensorHandle> m_CellToInputWeightsTensor; member in armnn::RefLstmWorkload
|
| H A D | RefQLstmWorkload.hpp | 36 std::unique_ptr<ScopedTensorHandle> m_CellToInputWeightsTensor; member in armnn::RefQLstmWorkload
|