Home
last modified time | relevance | path

Searched refs:m_OutputGateBiasTensor (Results 1 – 16 of 16) sorted by relevance

/external/armnn/src/backends/cl/workloads/
DClQuantizedLstmWorkload.cpp101 m_OutputGateBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClQuantizedLstmWorkload()
102 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in ClQuantizedLstmWorkload()
116m_OutputGateBiasTensor.get(), &cellStateInTensor, &outputStateInTensor, in ClQuantizedLstmWorkload()
130 InitializeArmComputeClTensorData(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias); in ClQuantizedLstmWorkload()
155 FreeTensorIfUnused(m_OutputGateBiasTensor); in FreeUnusedTensors()
DClQLstmWorkload.cpp47 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()
DClLstmFloatWorkload.cpp52 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()
DClQuantizedLstmWorkload.hpp42 std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor; member in armnn::ClQuantizedLstmWorkload
DClQLstmWorkload.hpp45 std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor; member in armnn::ClQLstmWorkload
DClLstmFloatWorkload.hpp41 std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor; member in armnn::ClLstmFloatWorkload
/external/armnn/src/backends/neon/workloads/
DNeonQuantizedLstmWorkload.cpp55 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()
DNeonQLstmWorkload.cpp47 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()
DNeonLstmFloatWorkload.cpp49 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()
DNeonQuantizedLstmWorkload.hpp39 std::unique_ptr<arm_compute::Tensor> m_OutputGateBiasTensor; member in armnn::NeonQuantizedLstmWorkload
DNeonQLstmWorkload.hpp45 std::unique_ptr<arm_compute::Tensor> m_OutputGateBiasTensor; member in armnn::NeonQLstmWorkload
DNeonLstmFloatWorkload.hpp42 std::unique_ptr<arm_compute::Tensor> m_OutputGateBiasTensor; member in armnn::NeonLstmFloatWorkload
/external/armnn/src/backends/reference/workloads/
DRefQLstmWorkload.hpp41 std::unique_ptr<ScopedCpuTensorHandle> m_OutputGateBiasTensor; member in armnn::RefQLstmWorkload
DRefLstmWorkload.hpp38 std::unique_ptr<ScopedCpuTensorHandle> m_OutputGateBiasTensor; member in armnn::RefLstmWorkload
DRefLstmWorkload.cpp32 , m_OutputGateBiasTensor (AssignScopedCpuTensorHandle(descriptor.m_OutputGateBias)) in RefLstmWorkload()
133 m_OutputGateBiasTensor->GetTensorInfo(), m_OutputGateBiasTensor->GetTensor<void>()); in Execute()
DRefQLstmWorkload.cpp35 , m_OutputGateBiasTensor (AssignScopedCpuTensorHandle(descriptor.m_OutputGateBias)) in RefQLstmWorkload()
265 outputGateBiasTensorInfo, m_OutputGateBiasTensor->GetTensor<void>()); in Execute()