| /external/armnn/src/backends/cl/workloads/ |
| D | ClQuantizedLstmWorkload.cpp | 99 m_CellBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClQuantizedLstmWorkload()
100 BuildArmComputeTensor(*m_CellBiasTensor, m_Data.m_CellBias->GetTensorInfo()); in ClQuantizedLstmWorkload()
120 m_CellBiasTensor.get(), in ClQuantizedLstmWorkload()
135 InitializeArmComputeClTensorData(*m_CellBiasTensor, m_Data.m_CellBias); in ClQuantizedLstmWorkload()
160 FreeTensorIfUnused(m_CellBiasTensor); in FreeUnusedTensors()
|
| D | ClQLstmWorkload.cpp | 52 m_CellBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload()
53 BuildArmComputeTensor(*m_CellBiasTensor, m_Data.m_CellBias->GetTensorInfo()); in ClQLstmWorkload()
173 m_CellBiasTensor.get(), in ClQLstmWorkload()
193 InitializeArmComputeClTensorData(*m_CellBiasTensor, m_Data.m_CellBias); in ClQLstmWorkload()
415 FreeTensorIfUnused(m_CellBiasTensor); in FreeUnusedTensors()
|
| D | ClLstmFloatWorkload.cpp | 58 m_CellBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload()
59 BuildArmComputeTensor(*m_CellBiasTensor, m_Data.m_CellBias->GetTensorInfo()); in ClLstmFloatWorkload()
176 … m_CellBiasTensor.get(), m_OutputGateBiasTensor.get(), &output_state_in, in ClLstmFloatWorkload()
191 InitializeArmComputeClTensorData(*m_CellBiasTensor, m_Data.m_CellBias); in ClLstmFloatWorkload()
385 FreeTensorIfUnused(m_CellBiasTensor); in FreeUnusedTensors()
|
| D | ClQuantizedLstmWorkload.hpp | 45 std::unique_ptr<arm_compute::CLTensor> m_CellBiasTensor; member in armnn::ClQuantizedLstmWorkload
|
| D | ClQLstmWorkload.hpp | 46 std::unique_ptr<arm_compute::CLTensor> m_CellBiasTensor; member in armnn::ClQLstmWorkload
|
| D | ClLstmFloatWorkload.hpp | 47 std::unique_ptr<arm_compute::CLTensor> m_CellBiasTensor; member in armnn::ClLstmFloatWorkload
|
| D | ClUnidirectionalSequenceLstmFloatWorkload.cpp | 174 m_CellBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClUnidirectionalSequenceLstmFloatWorkload()
175 BuildArmComputeTensor(*m_CellBiasTensor, m_Data.m_CellBias->GetTensorInfo()); in ClUnidirectionalSequenceLstmFloatWorkload()
333 m_CellBiasTensor.get(), in ClUnidirectionalSequenceLstmFloatWorkload()
358 InitializeArmComputeClTensorData(*m_CellBiasTensor, m_Data.m_CellBias); in ClUnidirectionalSequenceLstmFloatWorkload()
892 FreeTensorIfUnused(m_CellBiasTensor); in FreeUnusedTensors()
|
| D | ClUnidirectionalSequenceLstmFloatWorkload.hpp | 60 std::unique_ptr<arm_compute::CLTensor> m_CellBiasTensor; member in armnn::ClUnidirectionalSequenceLstmFloatWorkload
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonQuantizedLstmWorkload.cpp | 52 m_CellBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonQuantizedLstmWorkload()
53 BuildArmComputeTensor(*m_CellBiasTensor, m_Data.m_CellBias->GetTensorInfo()); in NeonQuantizedLstmWorkload()
76 m_CellBiasTensor.get(), in NeonQuantizedLstmWorkload()
113 InitializeArmComputeTensorData(*m_CellBiasTensor, in NeonQuantizedLstmWorkload()
204 FreeTensorIfUnused(m_CellBiasTensor); in FreeUnusedTensors()
|
| D | NeonQLstmWorkload.cpp | 50 m_CellBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload()
51 BuildArmComputeTensor(*m_CellBiasTensor, m_Data.m_CellBias->GetTensorInfo()); in NeonQLstmWorkload()
168 m_CellBiasTensor.get(), in NeonQLstmWorkload()
187 InitializeArmComputeTensorData(*m_CellBiasTensor, m_Data.m_CellBias); in NeonQLstmWorkload()
411 FreeTensorIfUnused(m_CellBiasTensor); in FreeUnusedTensors()
|
| D | NeonLstmFloatWorkload.cpp | 53 m_CellBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload()
54 BuildArmComputeTensor(*m_CellBiasTensor, m_Data.m_CellBias->GetTensorInfo()); in NeonLstmFloatWorkload()
170 … m_ForgetGateBiasTensor.get(), m_CellBiasTensor.get(), m_OutputGateBiasTensor.get(), in NeonLstmFloatWorkload()
191 InitializeArmComputeTensorData(*m_CellBiasTensor, in NeonLstmFloatWorkload()
406 FreeTensorIfUnused(m_CellBiasTensor); in FreeUnusedTensors()
|
| D | NeonQuantizedLstmWorkload.hpp | 38 std::unique_ptr<arm_compute::Tensor> m_CellBiasTensor; member in armnn::NeonQuantizedLstmWorkload
|
| D | NeonQLstmWorkload.hpp | 44 std::unique_ptr<arm_compute::Tensor> m_CellBiasTensor; member in armnn::NeonQLstmWorkload
|
| D | NeonLstmFloatWorkload.hpp | 45 std::unique_ptr<arm_compute::Tensor> m_CellBiasTensor; member in armnn::NeonLstmFloatWorkload
|
| D | NeonUnidirectionalSequenceLstmFloatWorkload.cpp | 172 m_CellBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmFloatWorkload()
173 BuildArmComputeTensor(*m_CellBiasTensor, m_Data.m_CellBias->GetTensorInfo()); in NeonUnidirectionalSequenceLstmFloatWorkload()
336 m_CellBiasTensor.get(), in NeonUnidirectionalSequenceLstmFloatWorkload()
361 InitializeArmComputeTensorData(*m_CellBiasTensor, m_Data.m_CellBias); in NeonUnidirectionalSequenceLstmFloatWorkload()
896 FreeTensorIfUnused(m_CellBiasTensor); in FreeUnusedTensors()
|
| D | NeonUnidirectionalSequenceLstmWorkload.cpp | 185 m_CellBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmWorkload()
186 BuildArmComputeTensor(*m_CellBiasTensor, m_Data.m_CellBias->GetTensorInfo()); in NeonUnidirectionalSequenceLstmWorkload()
323 m_CellBiasTensor.get(), in NeonUnidirectionalSequenceLstmWorkload()
342 InitializeArmComputeTensorData(*m_CellBiasTensor, m_Data.m_CellBias); in NeonUnidirectionalSequenceLstmWorkload()
869 FreeTensorIfUnused(m_CellBiasTensor); in FreeUnusedTensors()
|
| D | NeonUnidirectionalSequenceLstmWorkload.hpp | 56 std::unique_ptr<arm_compute::Tensor> m_CellBiasTensor; member in armnn::NeonUnidirectionalSequenceLstmWorkload
|
| D | NeonUnidirectionalSequenceLstmFloatWorkload.hpp | 58 std::unique_ptr<arm_compute::Tensor> m_CellBiasTensor; member in armnn::NeonUnidirectionalSequenceLstmFloatWorkload
|
| /external/armnn/src/backends/reference/workloads/ |
| D | RefLstmWorkload.cpp | 32 , m_CellBiasTensor (AssignScopedTensorHandle(descriptor.m_CellBias)) in RefLstmWorkload()
139 m_CellBiasTensor->GetTensorInfo(), m_CellBiasTensor->GetConstTensor<void>()); in Execute()
|
| D | RefUnidirectionalSequenceLstmWorkload.cpp | 36 , m_CellBiasTensor (AssignScopedTensorHandle(descriptor.m_CellBias)) in RefUnidirectionalSequenceLstmWorkload()
170 m_CellBiasTensor->GetTensorInfo(), m_CellBiasTensor->GetConstTensor<void>()); in Execute()
|
| D | RefQLstmWorkload.hpp | 42 std::unique_ptr<ScopedTensorHandle> m_CellBiasTensor; member in armnn::RefQLstmWorkload
|
| D | RefUnidirectionalSequenceLstmWorkload.hpp | 44 std::unique_ptr<ScopedTensorHandle> m_CellBiasTensor; member in armnn::RefUnidirectionalSequenceLstmWorkload
|
| D | RefLstmWorkload.hpp | 39 std::unique_ptr<ScopedTensorHandle> m_CellBiasTensor; member in armnn::RefLstmWorkload
|
| D | RefQLstmWorkload.cpp | 34 , m_CellBiasTensor (AssignScopedTensorHandle(descriptor.m_CellBias)) in RefQLstmWorkload()
276 cellGateBiasTensorInfo, m_CellBiasTensor->GetConstTensor<void>()); in Execute()
|