| /aosp_15_r20/external/armnn/src/backends/neon/workloads/ |
| H A D | NeonQuantizedLstmWorkload.cpp | 55 m_OutputGateBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonQuantizedLstmWorkload()
56 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in NeonQuantizedLstmWorkload()
77 m_OutputGateBiasTensor.get(), in NeonQuantizedLstmWorkload()
116 InitializeArmComputeTensorData(*m_OutputGateBiasTensor, in NeonQuantizedLstmWorkload()
205 FreeTensorIfUnused(m_OutputGateBiasTensor); in FreeUnusedTensors()
|
| H A D | NeonQLstmWorkload.cpp | 53 m_OutputGateBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload()
54 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in NeonQLstmWorkload()
169 m_OutputGateBiasTensor.get(), in NeonQLstmWorkload()
188 InitializeArmComputeTensorData(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias); in NeonQLstmWorkload()
412 FreeTensorIfUnused(m_OutputGateBiasTensor); in FreeUnusedTensors()
|
| H A D | NeonLstmFloatWorkload.cpp | 56 m_OutputGateBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload()
57 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in NeonLstmFloatWorkload()
170 … m_ForgetGateBiasTensor.get(), m_CellBiasTensor.get(), m_OutputGateBiasTensor.get(), in NeonLstmFloatWorkload()
193 InitializeArmComputeTensorData(*m_OutputGateBiasTensor, in NeonLstmFloatWorkload()
407 FreeTensorIfUnused(m_OutputGateBiasTensor); in FreeUnusedTensors()
|
| H A D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 175 m_OutputGateBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmFloatWorkload()
176 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in NeonUnidirectionalSequenceLstmFloatWorkload()
337 m_OutputGateBiasTensor.get(), in NeonUnidirectionalSequenceLstmFloatWorkload()
362 InitializeArmComputeTensorData(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias); in NeonUnidirectionalSequenceLstmFloatWorkload()
897 FreeTensorIfUnused(m_OutputGateBiasTensor); in FreeUnusedTensors()
|
| H A D | NeonUnidirectionalSequenceLstmWorkload.cpp | 188 m_OutputGateBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmWorkload()
189 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in NeonUnidirectionalSequenceLstmWorkload()
324 m_OutputGateBiasTensor.get(), in NeonUnidirectionalSequenceLstmWorkload()
343 InitializeArmComputeTensorData(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias); in NeonUnidirectionalSequenceLstmWorkload()
870 FreeTensorIfUnused(m_OutputGateBiasTensor); in FreeUnusedTensors()
|
| H A D | NeonQuantizedLstmWorkload.hpp | 39 std::unique_ptr<arm_compute::Tensor> m_OutputGateBiasTensor; member in armnn::NeonQuantizedLstmWorkload
|
| H A D | NeonQLstmWorkload.hpp | 45 std::unique_ptr<arm_compute::Tensor> m_OutputGateBiasTensor; member in armnn::NeonQLstmWorkload
|
| H A D | NeonLstmFloatWorkload.hpp | 46 std::unique_ptr<arm_compute::Tensor> m_OutputGateBiasTensor; member in armnn::NeonLstmFloatWorkload
|
| H A D | NeonUnidirectionalSequenceLstmWorkload.hpp | 57 std::unique_ptr<arm_compute::Tensor> m_OutputGateBiasTensor; member in armnn::NeonUnidirectionalSequenceLstmWorkload
|
| H A D | NeonUnidirectionalSequenceLstmFloatWorkload.hpp | 59 std::unique_ptr<arm_compute::Tensor> m_OutputGateBiasTensor; member in armnn::NeonUnidirectionalSequenceLstmFloatWorkload
|
| /aosp_15_r20/external/armnn/src/backends/cl/workloads/ |
| H A D | ClQuantizedLstmWorkload.cpp | 102 m_OutputGateBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClQuantizedLstmWorkload()
103 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in ClQuantizedLstmWorkload()
121 … m_OutputGateBiasTensor.get(), &cellStateInTensor, &outputStateInTensor, in ClQuantizedLstmWorkload()
136 InitializeArmComputeClTensorData(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias); in ClQuantizedLstmWorkload()
161 FreeTensorIfUnused(m_OutputGateBiasTensor); in FreeUnusedTensors()
|
| H A D | ClQLstmWorkload.cpp | 55 m_OutputGateBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload()
56 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in ClQLstmWorkload()
174 m_OutputGateBiasTensor.get(), in ClQLstmWorkload()
194 InitializeArmComputeClTensorData(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias); in ClQLstmWorkload()
416 FreeTensorIfUnused(m_OutputGateBiasTensor); in FreeUnusedTensors()
|
| H A D | ClLstmFloatWorkload.cpp | 61 m_OutputGateBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload()
62 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in ClLstmFloatWorkload()
176 … m_CellBiasTensor.get(), m_OutputGateBiasTensor.get(), &output_state_in, in ClLstmFloatWorkload()
192 InitializeArmComputeClTensorData(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias); in ClLstmFloatWorkload()
386 FreeTensorIfUnused(m_OutputGateBiasTensor); in FreeUnusedTensors()
|
| H A D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 177 m_OutputGateBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClUnidirectionalSequenceLstmFloatWorkload()
178 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in ClUnidirectionalSequenceLstmFloatWorkload()
334 m_OutputGateBiasTensor.get(), in ClUnidirectionalSequenceLstmFloatWorkload()
359 InitializeArmComputeClTensorData(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias); in ClUnidirectionalSequenceLstmFloatWorkload()
893 FreeTensorIfUnused(m_OutputGateBiasTensor); in FreeUnusedTensors()
|
| H A D | ClQuantizedLstmWorkload.hpp | 46 std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor; member in armnn::ClQuantizedLstmWorkload
|
| H A D | ClQLstmWorkload.hpp | 47 std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor; member in armnn::ClQLstmWorkload
|
| H A D | ClLstmFloatWorkload.hpp | 48 std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor; member in armnn::ClLstmFloatWorkload
|
| H A D | ClUnidirectionalSequenceLstmFloatWorkload.hpp | 61 std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor; member in armnn::ClUnidirectionalSequenceLstmFloatWorkload
|
| /aosp_15_r20/external/armnn/src/backends/reference/workloads/ |
| H A D | RefLstmWorkload.cpp | 33 , m_OutputGateBiasTensor (AssignScopedTensorHandle(descriptor.m_OutputGateBias)) in RefLstmWorkload()
141 m_OutputGateBiasTensor->GetTensorInfo(), m_OutputGateBiasTensor->GetConstTensor<void>()); in Execute()
|
| H A D | RefUnidirectionalSequenceLstmWorkload.cpp | 37 , m_OutputGateBiasTensor (AssignScopedTensorHandle(descriptor.m_OutputGateBias)) in RefUnidirectionalSequenceLstmWorkload()
172 m_OutputGateBiasTensor->GetTensorInfo(), m_OutputGateBiasTensor->GetConstTensor<void>()); in Execute()
|
| H A D | RefUnidirectionalSequenceLstmWorkload.hpp | 45 std::unique_ptr<ScopedTensorHandle> m_OutputGateBiasTensor; member in armnn::RefUnidirectionalSequenceLstmWorkload
|
| H A D | RefLstmWorkload.hpp | 40 std::unique_ptr<ScopedTensorHandle> m_OutputGateBiasTensor; member in armnn::RefLstmWorkload
|
| H A D | RefQLstmWorkload.hpp | 43 std::unique_ptr<ScopedTensorHandle> m_OutputGateBiasTensor; member in armnn::RefQLstmWorkload
|
| H A D | RefQLstmWorkload.cpp | 35 , m_OutputGateBiasTensor (AssignScopedTensorHandle(descriptor.m_OutputGateBias)) in RefQLstmWorkload()
281 outputGateBiasTensorInfo, m_OutputGateBiasTensor->GetConstTensor<void>()); in Execute()
|