/external/armnn/src/backends/cl/workloads/ |
D | ClQuantizedLstmWorkload.cpp | 101 m_OutputGateBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClQuantizedLstmWorkload() 102 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in ClQuantizedLstmWorkload() 116 … m_OutputGateBiasTensor.get(), &cellStateInTensor, &outputStateInTensor, in ClQuantizedLstmWorkload() 130 InitializeArmComputeClTensorData(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias); in ClQuantizedLstmWorkload() 155 FreeTensorIfUnused(m_OutputGateBiasTensor); in FreeUnusedTensors()
|
D | ClQLstmWorkload.cpp | 47 m_OutputGateBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload() 48 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in ClQLstmWorkload() 163 m_OutputGateBiasTensor.get(), in ClQLstmWorkload() 182 InitializeArmComputeClTensorData(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias); in ClQLstmWorkload() 403 FreeTensorIfUnused(m_OutputGateBiasTensor); in FreeUnusedTensors()
|
D | ClLstmFloatWorkload.cpp | 52 m_OutputGateBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload() 53 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in ClLstmFloatWorkload() 191 … m_ForgetGateBiasTensor.get(), m_CellBiasTensor.get(), m_OutputGateBiasTensor.get(), in ClLstmFloatWorkload() 206 InitializeArmComputeClTensorData(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias); in ClLstmFloatWorkload() 427 FreeTensorIfUnused(m_OutputGateBiasTensor); in FreeUnusedTensors()
|
D | ClQuantizedLstmWorkload.hpp | 42 std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor; member in armnn::ClQuantizedLstmWorkload
|
D | ClQLstmWorkload.hpp | 45 std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor; member in armnn::ClQLstmWorkload
|
D | ClLstmFloatWorkload.hpp | 41 std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor; member in armnn::ClLstmFloatWorkload
|
/external/armnn/src/backends/neon/workloads/ |
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() 204 FreeTensorIfUnused(m_OutputGateBiasTensor); in FreeUnusedTensors()
|
D | NeonQLstmWorkload.cpp | 47 m_OutputGateBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload() 48 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in NeonQLstmWorkload() 163 m_OutputGateBiasTensor.get(), in NeonQLstmWorkload() 182 InitializeArmComputeTensorData(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias); in NeonQLstmWorkload() 405 FreeTensorIfUnused(m_OutputGateBiasTensor); in FreeUnusedTensors()
|
D | NeonLstmFloatWorkload.cpp | 49 m_OutputGateBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload() 50 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in NeonLstmFloatWorkload() 191 … m_ForgetGateBiasTensor.get(), m_CellBiasTensor.get(), m_OutputGateBiasTensor.get(), in NeonLstmFloatWorkload() 214 InitializeArmComputeTensorData(*m_OutputGateBiasTensor, in NeonLstmFloatWorkload() 450 FreeTensorIfUnused(m_OutputGateBiasTensor); in FreeUnusedTensors()
|
D | NeonQuantizedLstmWorkload.hpp | 39 std::unique_ptr<arm_compute::Tensor> m_OutputGateBiasTensor; member in armnn::NeonQuantizedLstmWorkload
|
D | NeonQLstmWorkload.hpp | 45 std::unique_ptr<arm_compute::Tensor> m_OutputGateBiasTensor; member in armnn::NeonQLstmWorkload
|
D | NeonLstmFloatWorkload.hpp | 42 std::unique_ptr<arm_compute::Tensor> m_OutputGateBiasTensor; member in armnn::NeonLstmFloatWorkload
|
/external/armnn/src/backends/reference/workloads/ |
D | RefQLstmWorkload.hpp | 41 std::unique_ptr<ScopedCpuTensorHandle> m_OutputGateBiasTensor; member in armnn::RefQLstmWorkload
|
D | RefLstmWorkload.hpp | 38 std::unique_ptr<ScopedCpuTensorHandle> m_OutputGateBiasTensor; member in armnn::RefLstmWorkload
|
D | RefLstmWorkload.cpp | 32 , m_OutputGateBiasTensor (AssignScopedCpuTensorHandle(descriptor.m_OutputGateBias)) in RefLstmWorkload() 133 m_OutputGateBiasTensor->GetTensorInfo(), m_OutputGateBiasTensor->GetTensor<void>()); in Execute()
|
D | RefQLstmWorkload.cpp | 35 , m_OutputGateBiasTensor (AssignScopedCpuTensorHandle(descriptor.m_OutputGateBias)) in RefQLstmWorkload() 265 outputGateBiasTensorInfo, m_OutputGateBiasTensor->GetTensor<void>()); in Execute()
|