| /aosp_15_r20/external/armnn/src/backends/neon/workloads/ |
| H A D | NeonQuantizedLstmWorkload.cpp | 46 m_InputGateBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonQuantizedLstmWorkload()
47 BuildArmComputeTensor(*m_InputGateBiasTensor, m_Data.m_InputGateBias->GetTensorInfo()); in NeonQuantizedLstmWorkload()
74 m_InputGateBiasTensor.get(), in NeonQuantizedLstmWorkload()
107 InitializeArmComputeTensorData(*m_InputGateBiasTensor, in NeonQuantizedLstmWorkload()
202 FreeTensorIfUnused(m_InputGateBiasTensor); in FreeUnusedTensors()
|
| H A D | NeonQLstmWorkload.cpp | 129 m_InputGateBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload()
130 BuildArmComputeTensor(*m_InputGateBiasTensor, m_Data.m_InputGateBias->GetTensorInfo()); in NeonQLstmWorkload()
137 m_InputGateBiasTensor.get()); in NeonQLstmWorkload()
195 InitializeArmComputeTensorData(*m_InputGateBiasTensor, m_Data.m_InputGateBias); in NeonQLstmWorkload()
409 FreeTensorIfUnused(m_InputGateBiasTensor); in FreeUnusedTensors()
|
| H A D | NeonLstmFloatWorkload.cpp | 74 m_InputGateBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload()
75 BuildArmComputeTensor(*m_InputGateBiasTensor, m_Data.m_InputGateBias->GetTensorInfo()); in NeonLstmFloatWorkload()
80 m_InputGateBiasTensor.get()); in NeonLstmFloatWorkload()
207 InitializeArmComputeTensorData(*m_InputGateBiasTensor, in NeonLstmFloatWorkload()
404 FreeTensorIfUnused(m_InputGateBiasTensor); in FreeUnusedTensors()
|
| H A D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 193 m_InputGateBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmFloatWorkload()
194 BuildArmComputeTensor(*m_InputGateBiasTensor, m_Data.m_InputGateBias->GetTensorInfo()); in NeonUnidirectionalSequenceLstmFloatWorkload()
199 m_InputGateBiasTensor.get()); in NeonUnidirectionalSequenceLstmFloatWorkload()
372 InitializeArmComputeTensorData(*m_InputGateBiasTensor, m_Data.m_InputGateBias); in NeonUnidirectionalSequenceLstmFloatWorkload()
894 FreeTensorIfUnused(m_InputGateBiasTensor); in FreeUnusedTensors()
|
| H A D | NeonUnidirectionalSequenceLstmWorkload.cpp | 206 m_InputGateBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmWorkload()
207 BuildArmComputeTensor(*m_InputGateBiasTensor, m_Data.m_InputGateBias->GetTensorInfo()); in NeonUnidirectionalSequenceLstmWorkload()
211 m_InputGateBiasTensor.get()); in NeonUnidirectionalSequenceLstmWorkload()
353 InitializeArmComputeTensorData(*m_InputGateBiasTensor, m_Data.m_InputGateBias); in NeonUnidirectionalSequenceLstmWorkload()
867 FreeTensorIfUnused(m_InputGateBiasTensor); in FreeUnusedTensors()
|
| H A D | NeonQuantizedLstmWorkload.hpp | 36 std::unique_ptr<arm_compute::Tensor> m_InputGateBiasTensor; member in armnn::NeonQuantizedLstmWorkload
|
| H A D | NeonQLstmWorkload.hpp | 42 std::unique_ptr<arm_compute::Tensor> m_InputGateBiasTensor; member in armnn::NeonQLstmWorkload
|
| H A D | NeonLstmFloatWorkload.hpp | 43 std::unique_ptr<arm_compute::Tensor> m_InputGateBiasTensor; member in armnn::NeonLstmFloatWorkload
|
| H A D | NeonUnidirectionalSequenceLstmWorkload.hpp | 54 std::unique_ptr<arm_compute::Tensor> m_InputGateBiasTensor; member in armnn::NeonUnidirectionalSequenceLstmWorkload
|
| H A D | NeonUnidirectionalSequenceLstmFloatWorkload.hpp | 56 std::unique_ptr<arm_compute::Tensor> m_InputGateBiasTensor; member in armnn::NeonUnidirectionalSequenceLstmFloatWorkload
|
| /aosp_15_r20/external/armnn/src/backends/cl/workloads/ |
| H A D | ClQuantizedLstmWorkload.cpp | 93 m_InputGateBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClQuantizedLstmWorkload()
94 BuildArmComputeTensor(*m_InputGateBiasTensor, m_Data.m_InputGateBias->GetTensorInfo()); in ClQuantizedLstmWorkload()
119 m_InputGateBiasTensor.get(), m_ForgetGateBiasTensor.get(), in ClQuantizedLstmWorkload()
133 InitializeArmComputeClTensorData(*m_InputGateBiasTensor, m_Data.m_InputGateBias); in ClQuantizedLstmWorkload()
158 FreeTensorIfUnused(m_InputGateBiasTensor); in FreeUnusedTensors()
|
| H A D | ClQLstmWorkload.cpp | 131 m_InputGateBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload()
132 BuildArmComputeTensor(*m_InputGateBiasTensor, m_Data.m_InputGateBias->GetTensorInfo()); in ClQLstmWorkload()
139 m_InputGateBiasTensor.get()); in ClQLstmWorkload()
201 InitializeArmComputeClTensorData(*m_InputGateBiasTensor, m_Data.m_InputGateBias); in ClQLstmWorkload()
413 FreeTensorIfUnused(m_InputGateBiasTensor); in FreeUnusedTensors()
|
| H A D | ClLstmFloatWorkload.cpp | 79 m_InputGateBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload()
80 BuildArmComputeTensor(*m_InputGateBiasTensor, m_Data.m_InputGateBias->GetTensorInfo()); in ClLstmFloatWorkload()
85 m_InputGateBiasTensor.get()); in ClLstmFloatWorkload()
202 InitializeArmComputeClTensorData(*m_InputGateBiasTensor, m_Data.m_InputGateBias); in ClLstmFloatWorkload()
383 FreeTensorIfUnused(m_InputGateBiasTensor); in FreeUnusedTensors()
|
| H A D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 195 m_InputGateBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClUnidirectionalSequenceLstmFloatWorkload()
196 BuildArmComputeTensor(*m_InputGateBiasTensor, m_Data.m_InputGateBias->GetTensorInfo()); in ClUnidirectionalSequenceLstmFloatWorkload()
201 m_InputGateBiasTensor.get()); in ClUnidirectionalSequenceLstmFloatWorkload()
369 InitializeArmComputeClTensorData(*m_InputGateBiasTensor, m_Data.m_InputGateBias); in ClUnidirectionalSequenceLstmFloatWorkload()
890 FreeTensorIfUnused(m_InputGateBiasTensor); in FreeUnusedTensors()
|
| H A D | ClQuantizedLstmWorkload.hpp | 43 std::unique_ptr<arm_compute::CLTensor> m_InputGateBiasTensor; member in armnn::ClQuantizedLstmWorkload
|
| H A D | ClQLstmWorkload.hpp | 44 std::unique_ptr<arm_compute::CLTensor> m_InputGateBiasTensor; member in armnn::ClQLstmWorkload
|
| H A D | ClLstmFloatWorkload.hpp | 45 std::unique_ptr<arm_compute::CLTensor> m_InputGateBiasTensor; member in armnn::ClLstmFloatWorkload
|
| H A D | ClUnidirectionalSequenceLstmFloatWorkload.hpp | 58 std::unique_ptr<arm_compute::CLTensor> m_InputGateBiasTensor; member in armnn::ClUnidirectionalSequenceLstmFloatWorkload
|
| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | RefLstmWorkload.cpp | 30 , m_InputGateBiasTensor (AssignScopedTensorHandle(descriptor.m_InputGateBias)) in RefLstmWorkload()
178 m_InputGateBiasTensor->GetTensorInfo(), m_InputGateBiasTensor->GetConstTensor<void>()); in Execute()
|
| H A D | RefUnidirectionalSequenceLstmWorkload.cpp | 34 , m_InputGateBiasTensor (AssignScopedTensorHandle(descriptor.m_InputGateBias)) in RefUnidirectionalSequenceLstmWorkload()
206 m_InputGateBiasTensor->GetTensorInfo(), m_InputGateBiasTensor->GetConstTensor<void>()); in Execute()
|
| H A D | RefUnidirectionalSequenceLstmWorkload.hpp | 42 std::unique_ptr<ScopedTensorHandle> m_InputGateBiasTensor; member in armnn::RefUnidirectionalSequenceLstmWorkload
|
| H A D | RefLstmWorkload.hpp | 37 std::unique_ptr<ScopedTensorHandle> m_InputGateBiasTensor; member in armnn::RefLstmWorkload
|
| H A D | RefQLstmWorkload.hpp | 40 std::unique_ptr<ScopedTensorHandle> m_InputGateBiasTensor; member in armnn::RefQLstmWorkload
|
| H A D | RefQLstmWorkload.cpp | 32 , m_InputGateBiasTensor (AssignScopedTensorHandle(descriptor.m_InputGateBias)) in RefQLstmWorkload()
255 inputGateBiasTensorInfo, m_InputGateBiasTensor->GetConstTensor<void>()); in Execute()
|