| /aosp_15_r20/external/armnn/src/backends/neon/workloads/ |
| H A D | NeonQuantizedLstmWorkload.cpp | 28 m_InputToCellWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQuantizedLstmWorkload()
29 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in NeonQuantizedLstmWorkload()
68 m_InputToCellWeightsTensor.get(), in NeonQuantizedLstmWorkload()
89 InitializeArmComputeTensorData(*m_InputToCellWeightsTensor, in NeonQuantizedLstmWorkload()
196 FreeTensorIfUnused(m_InputToCellWeightsTensor); in FreeUnusedTensors()
|
| H A D | NeonQLstmWorkload.cpp | 32 m_InputToCellWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload()
33 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in NeonQLstmWorkload()
162 m_InputToCellWeightsTensor.get(), in NeonQLstmWorkload()
179 InitializeArmComputeTensorData(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights); in NeonQLstmWorkload()
397 FreeTensorIfUnused(m_InputToCellWeightsTensor); in FreeUnusedTensors()
|
| H A D | NeonLstmFloatWorkload.cpp | 35 m_InputToCellWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload()
36 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in NeonLstmFloatWorkload()
167 …m_LstmLayer.configure(&input, m_InputToForgetWeightsTensor.get(), m_InputToCellWeightsTensor.get(), in NeonLstmFloatWorkload()
179 InitializeArmComputeTensorData(*m_InputToCellWeightsTensor, in NeonLstmFloatWorkload()
395 FreeTensorIfUnused(m_InputToCellWeightsTensor); in FreeUnusedTensors()
|
| H A D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 154 m_InputToCellWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmFloatWorkload()
155 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in NeonUnidirectionalSequenceLstmFloatWorkload()
330 m_InputToCellWeightsTensor.get(), in NeonUnidirectionalSequenceLstmFloatWorkload()
355 InitializeArmComputeTensorData(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights); in NeonUnidirectionalSequenceLstmFloatWorkload()
885 FreeTensorIfUnused(m_InputToCellWeightsTensor); in FreeUnusedTensors()
|
| H A D | NeonUnidirectionalSequenceLstmWorkload.cpp | 167 m_InputToCellWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmWorkload()
168 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in NeonUnidirectionalSequenceLstmWorkload()
317 m_InputToCellWeightsTensor.get(), in NeonUnidirectionalSequenceLstmWorkload()
336 InitializeArmComputeTensorData(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights); in NeonUnidirectionalSequenceLstmWorkload()
858 FreeTensorIfUnused(m_InputToCellWeightsTensor); in FreeUnusedTensors()
|
| H A D | NeonQuantizedLstmWorkload.hpp | 30 std::unique_ptr<arm_compute::Tensor> m_InputToCellWeightsTensor; member in armnn::NeonQuantizedLstmWorkload
|
| H A D | NeonQLstmWorkload.hpp | 30 std::unique_ptr<arm_compute::Tensor> m_InputToCellWeightsTensor; member in armnn::NeonQLstmWorkload
|
| H A D | NeonLstmFloatWorkload.hpp | 34 std::unique_ptr<arm_compute::Tensor> m_InputToCellWeightsTensor; member in armnn::NeonLstmFloatWorkload
|
| H A D | NeonUnidirectionalSequenceLstmWorkload.hpp | 45 std::unique_ptr<arm_compute::Tensor> m_InputToCellWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmWorkload
|
| H A D | NeonUnidirectionalSequenceLstmFloatWorkload.hpp | 47 std::unique_ptr<arm_compute::Tensor> m_InputToCellWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmFloatWorkload
|
| /aosp_15_r20/external/armnn/src/backends/cl/workloads/ |
| H A D | ClQuantizedLstmWorkload.cpp | 75 m_InputToCellWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQuantizedLstmWorkload()
76 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in ClQuantizedLstmWorkload()
116 … m_InputToCellWeightsTensor.get(), m_InputToOutputWeightsTensor.get(), in ClQuantizedLstmWorkload()
127 … InitializeArmComputeClTensorData(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights); in ClQuantizedLstmWorkload()
152 FreeTensorIfUnused(m_InputToCellWeightsTensor); in FreeUnusedTensors()
|
| H A D | ClQLstmWorkload.cpp | 34 m_InputToCellWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload()
35 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in ClQLstmWorkload()
167 m_InputToCellWeightsTensor.get(), in ClQLstmWorkload()
185 InitializeArmComputeClTensorData(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights); in ClQLstmWorkload()
401 FreeTensorIfUnused(m_InputToCellWeightsTensor); in FreeUnusedTensors()
|
| H A D | ClLstmFloatWorkload.cpp | 40 m_InputToCellWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload()
41 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in ClLstmFloatWorkload()
173 m_InputToCellWeightsTensor.get(), m_InputToOutputWeightsTensor.get(), in ClLstmFloatWorkload()
185 … InitializeArmComputeClTensorData(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights); in ClLstmFloatWorkload()
374 FreeTensorIfUnused(m_InputToCellWeightsTensor); in FreeUnusedTensors()
|
| H A D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 156 m_InputToCellWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClUnidirectionalSequenceLstmFloatWorkload()
157 … BuildArmComputeTensor(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights->GetTensorInfo()); in ClUnidirectionalSequenceLstmFloatWorkload()
327 m_InputToCellWeightsTensor.get(), in ClUnidirectionalSequenceLstmFloatWorkload()
352 InitializeArmComputeClTensorData(*m_InputToCellWeightsTensor, m_Data.m_InputToCellWeights); in ClUnidirectionalSequenceLstmFloatWorkload()
881 FreeTensorIfUnused(m_InputToCellWeightsTensor); in FreeUnusedTensors()
|
| H A D | ClQuantizedLstmWorkload.hpp | 37 std::unique_ptr<arm_compute::CLTensor> m_InputToCellWeightsTensor; member in armnn::ClQuantizedLstmWorkload
|
| H A D | ClQLstmWorkload.hpp | 32 std::unique_ptr<arm_compute::CLTensor> m_InputToCellWeightsTensor; member in armnn::ClQLstmWorkload
|
| H A D | ClLstmFloatWorkload.hpp | 36 std::unique_ptr<arm_compute::CLTensor> m_InputToCellWeightsTensor; member in armnn::ClLstmFloatWorkload
|
| H A D | ClUnidirectionalSequenceLstmFloatWorkload.hpp | 49 std::unique_ptr<arm_compute::CLTensor> m_InputToCellWeightsTensor; member in armnn::ClUnidirectionalSequenceLstmFloatWorkload
|
| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | RefLstmWorkload.cpp | 21 , m_InputToCellWeightsTensor (AssignScopedTensorHandle(descriptor.m_InputToCellWeights)) in RefLstmWorkload()
123 … m_InputToCellWeightsTensor->GetTensorInfo(), m_InputToCellWeightsTensor->GetConstTensor<void>()); in Execute()
|
| H A D | RefUnidirectionalSequenceLstmWorkload.cpp | 25 , m_InputToCellWeightsTensor (AssignScopedTensorHandle(descriptor.m_InputToCellWeights)) in RefUnidirectionalSequenceLstmWorkload()
154 … m_InputToCellWeightsTensor->GetTensorInfo(), m_InputToCellWeightsTensor->GetConstTensor<void>()); in Execute()
|
| H A D | RefQLstmWorkload.cpp | 20 … , m_InputToCellWeightsTensor (AssignScopedTensorHandle(descriptor.m_InputToCellWeights)) in RefQLstmWorkload()
118 … m_InputToCellWeightsTensor->GetTensorInfo(), m_InputToCellWeightsTensor->GetConstTensor<void>()); in Execute()
|
| H A D | RefUnidirectionalSequenceLstmWorkload.hpp | 33 std::unique_ptr<ScopedTensorHandle> m_InputToCellWeightsTensor; member in armnn::RefUnidirectionalSequenceLstmWorkload
|
| H A D | RefLstmWorkload.hpp | 28 std::unique_ptr<ScopedTensorHandle> m_InputToCellWeightsTensor; member in armnn::RefLstmWorkload
|
| H A D | RefQLstmWorkload.hpp | 28 std::unique_ptr<ScopedTensorHandle> m_InputToCellWeightsTensor; member in armnn::RefQLstmWorkload
|