Home
last modified time | relevance | path

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

/external/armnn/src/backends/cl/workloads/
DClQuantizedLstmWorkload.cpp92 m_InputGateBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClQuantizedLstmWorkload()
93 BuildArmComputeTensor(*m_InputGateBiasTensor, m_Data.m_InputGateBias->GetTensorInfo()); in ClQuantizedLstmWorkload()
115m_InputGateBiasTensor.get(), m_ForgetGateBiasTensor.get(), m_CellBiasTensor.get(), in ClQuantizedLstmWorkload()
127 InitializeArmComputeClTensorData(*m_InputGateBiasTensor, m_Data.m_InputGateBias); in ClQuantizedLstmWorkload()
152 FreeTensorIfUnused(m_InputGateBiasTensor); in FreeUnusedTensors()
DClQLstmWorkload.cpp123 m_InputGateBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload()
124 BuildArmComputeTensor(*m_InputGateBiasTensor, m_Data.m_InputGateBias->GetTensorInfo()); in ClQLstmWorkload()
131 m_InputGateBiasTensor.get()); in ClQLstmWorkload()
189 InitializeArmComputeClTensorData(*m_InputGateBiasTensor, m_Data.m_InputGateBias); in ClQLstmWorkload()
400 FreeTensorIfUnused(m_InputGateBiasTensor); in FreeUnusedTensors()
DClLstmFloatWorkload.cpp70 m_InputGateBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload()
71 BuildArmComputeTensor(*m_InputGateBiasTensor, m_Data.m_InputGateBias->GetTensorInfo()); in ClLstmFloatWorkload()
76 m_InputGateBiasTensor.get()); in ClLstmFloatWorkload()
216 InitializeArmComputeClTensorData(*m_InputGateBiasTensor, m_Data.m_InputGateBias); in ClLstmFloatWorkload()
424 FreeTensorIfUnused(m_InputGateBiasTensor); in FreeUnusedTensors()
DClQuantizedLstmWorkload.hpp39 std::unique_ptr<arm_compute::CLTensor> m_InputGateBiasTensor; member in armnn::ClQuantizedLstmWorkload
DClQLstmWorkload.hpp42 std::unique_ptr<arm_compute::CLTensor> m_InputGateBiasTensor; member in armnn::ClQLstmWorkload
DClLstmFloatWorkload.hpp38 std::unique_ptr<arm_compute::CLTensor> m_InputGateBiasTensor; member in armnn::ClLstmFloatWorkload
/external/armnn/src/backends/neon/workloads/
DNeonQuantizedLstmWorkload.cpp46 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()
201 FreeTensorIfUnused(m_InputGateBiasTensor); in FreeUnusedTensors()
DNeonQLstmWorkload.cpp123 m_InputGateBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload()
124 BuildArmComputeTensor(*m_InputGateBiasTensor, m_Data.m_InputGateBias->GetTensorInfo()); in NeonQLstmWorkload()
131 m_InputGateBiasTensor.get()); in NeonQLstmWorkload()
189 InitializeArmComputeTensorData(*m_InputGateBiasTensor, m_Data.m_InputGateBias); in NeonQLstmWorkload()
402 FreeTensorIfUnused(m_InputGateBiasTensor); in FreeUnusedTensors()
DNeonLstmFloatWorkload.cpp67 m_InputGateBiasTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload()
68 BuildArmComputeTensor(*m_InputGateBiasTensor, m_Data.m_InputGateBias->GetTensorInfo()); in NeonLstmFloatWorkload()
73 m_InputGateBiasTensor.get()); in NeonLstmFloatWorkload()
228 InitializeArmComputeTensorData(*m_InputGateBiasTensor, in NeonLstmFloatWorkload()
447 FreeTensorIfUnused(m_InputGateBiasTensor); in FreeUnusedTensors()
DNeonQuantizedLstmWorkload.hpp36 std::unique_ptr<arm_compute::Tensor> m_InputGateBiasTensor; member in armnn::NeonQuantizedLstmWorkload
DNeonQLstmWorkload.hpp42 std::unique_ptr<arm_compute::Tensor> m_InputGateBiasTensor; member in armnn::NeonQLstmWorkload
DNeonLstmFloatWorkload.hpp39 std::unique_ptr<arm_compute::Tensor> m_InputGateBiasTensor; member in armnn::NeonLstmFloatWorkload
/external/armnn/src/backends/reference/workloads/
DRefQLstmWorkload.hpp38 std::unique_ptr<ScopedCpuTensorHandle> m_InputGateBiasTensor; member in armnn::RefQLstmWorkload
DRefLstmWorkload.hpp35 std::unique_ptr<ScopedCpuTensorHandle> m_InputGateBiasTensor; member in armnn::RefLstmWorkload
DRefLstmWorkload.cpp29 , m_InputGateBiasTensor (AssignScopedCpuTensorHandle(descriptor.m_InputGateBias)) in RefLstmWorkload()
167 m_InputGateBiasTensor->GetTensorInfo(), m_InputGateBiasTensor->GetTensor<void>()); in Execute()
DRefQLstmWorkload.cpp32 , m_InputGateBiasTensor (AssignScopedCpuTensorHandle(descriptor.m_InputGateBias)) in RefQLstmWorkload()
241 inputGateBiasTensorInfo, m_InputGateBiasTensor->GetTensor<void>()); in Execute()