Home
last modified time | relevance | path

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

/external/armnn/src/backends/cl/workloads/
DClQuantizedLstmWorkload.cpp102 m_OutputGateBiasTensor = std::make_unique<arm_compute::CLTensor>(); in ClQuantizedLstmWorkload()
103 BuildArmComputeTensor(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias->GetTensorInfo()); in ClQuantizedLstmWorkload()
121m_OutputGateBiasTensor.get(), &cellStateInTensor, &outputStateInTensor, in ClQuantizedLstmWorkload()
136 InitializeArmComputeClTensorData(*m_OutputGateBiasTensor, m_Data.m_OutputGateBias); in ClQuantizedLstmWorkload()
161 FreeTensorIfUnused(m_OutputGateBiasTensor); in FreeUnusedTensors()
DClQLstmWorkload.cpp55 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()
DClLstmFloatWorkload.cpp61 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()
DClQuantizedLstmWorkload.hpp46 std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor; member in armnn::ClQuantizedLstmWorkload
DClUnidirectionalSequenceLstmFloatWorkload.cpp177 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()
DClQLstmWorkload.hpp47 std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor; member in armnn::ClQLstmWorkload
DClLstmFloatWorkload.hpp48 std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor; member in armnn::ClLstmFloatWorkload
DClUnidirectionalSequenceLstmFloatWorkload.hpp61 std::unique_ptr<arm_compute::CLTensor> m_OutputGateBiasTensor; member in armnn::ClUnidirectionalSequenceLstmFloatWorkload
/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()
205 FreeTensorIfUnused(m_OutputGateBiasTensor); in FreeUnusedTensors()
DNeonQLstmWorkload.cpp53 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()
DNeonLstmFloatWorkload.cpp56 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()
DNeonQuantizedLstmWorkload.hpp39 std::unique_ptr<arm_compute::Tensor> m_OutputGateBiasTensor; member in armnn::NeonQuantizedLstmWorkload
DNeonUnidirectionalSequenceLstmFloatWorkload.cpp175 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()
DNeonUnidirectionalSequenceLstmWorkload.cpp188 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()
DNeonQLstmWorkload.hpp45 std::unique_ptr<arm_compute::Tensor> m_OutputGateBiasTensor; member in armnn::NeonQLstmWorkload
DNeonLstmFloatWorkload.hpp46 std::unique_ptr<arm_compute::Tensor> m_OutputGateBiasTensor; member in armnn::NeonLstmFloatWorkload
DNeonUnidirectionalSequenceLstmFloatWorkload.hpp59 std::unique_ptr<arm_compute::Tensor> m_OutputGateBiasTensor; member in armnn::NeonUnidirectionalSequenceLstmFloatWorkload
DNeonUnidirectionalSequenceLstmWorkload.hpp57 std::unique_ptr<arm_compute::Tensor> m_OutputGateBiasTensor; member in armnn::NeonUnidirectionalSequenceLstmWorkload
/external/armnn/src/backends/reference/workloads/
DRefLstmWorkload.cpp33 , m_OutputGateBiasTensor (AssignScopedTensorHandle(descriptor.m_OutputGateBias)) in RefLstmWorkload()
141 m_OutputGateBiasTensor->GetTensorInfo(), m_OutputGateBiasTensor->GetConstTensor<void>()); in Execute()
DRefUnidirectionalSequenceLstmWorkload.cpp37 , m_OutputGateBiasTensor (AssignScopedTensorHandle(descriptor.m_OutputGateBias)) in RefUnidirectionalSequenceLstmWorkload()
172 m_OutputGateBiasTensor->GetTensorInfo(), m_OutputGateBiasTensor->GetConstTensor<void>()); in Execute()
DRefQLstmWorkload.hpp43 std::unique_ptr<ScopedTensorHandle> m_OutputGateBiasTensor; member in armnn::RefQLstmWorkload
DRefUnidirectionalSequenceLstmWorkload.hpp45 std::unique_ptr<ScopedTensorHandle> m_OutputGateBiasTensor; member in armnn::RefUnidirectionalSequenceLstmWorkload
DRefLstmWorkload.hpp40 std::unique_ptr<ScopedTensorHandle> m_OutputGateBiasTensor; member in armnn::RefLstmWorkload
DRefQLstmWorkload.cpp35 , m_OutputGateBiasTensor (AssignScopedTensorHandle(descriptor.m_OutputGateBias)) in RefQLstmWorkload()
281 outputGateBiasTensorInfo, m_OutputGateBiasTensor->GetConstTensor<void>()); in Execute()