Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonQuantizedLstmWorkload.cpp37 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQuantizedLstmWorkload()
38 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in NeonQuantizedLstmWorkload()
71 m_RecurrentToForgetWeightsTensor.get(), in NeonQuantizedLstmWorkload()
98 InitializeArmComputeTensorData(*m_RecurrentToForgetWeightsTensor, in NeonQuantizedLstmWorkload()
199 FreeTensorIfUnused(m_RecurrentToForgetWeightsTensor); in FreeUnusedTensors()
H A DNeonQLstmWorkload.cpp38 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload()
39 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in NeonQLstmWorkload()
164 m_RecurrentToForgetWeightsTensor.get(), in NeonQLstmWorkload()
182 …InitializeArmComputeTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeight… in NeonQLstmWorkload()
401 FreeTensorIfUnused(m_RecurrentToForgetWeightsTensor); in FreeUnusedTensors()
H A DNeonLstmFloatWorkload.cpp41 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload()
42 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in NeonLstmFloatWorkload()
168 … m_InputToOutputWeightsTensor.get(), m_RecurrentToForgetWeightsTensor.get(), in NeonLstmFloatWorkload()
183 InitializeArmComputeTensorData(*m_RecurrentToForgetWeightsTensor, in NeonLstmFloatWorkload()
398 FreeTensorIfUnused(m_RecurrentToForgetWeightsTensor); in FreeUnusedTensors()
H A DNeonUnidirectionalSequenceLstmFloatWorkload.cpp160 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmFloatWorkload()
161 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in NeonUnidirectionalSequenceLstmFloatWorkload()
332 m_RecurrentToForgetWeightsTensor.get(), in NeonUnidirectionalSequenceLstmFloatWorkload()
357 …InitializeArmComputeTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeight… in NeonUnidirectionalSequenceLstmFloatWorkload()
888 FreeTensorIfUnused(m_RecurrentToForgetWeightsTensor); in FreeUnusedTensors()
H A DNeonUnidirectionalSequenceLstmWorkload.cpp173 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmWorkload()
174 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in NeonUnidirectionalSequenceLstmWorkload()
319 m_RecurrentToForgetWeightsTensor.get(), in NeonUnidirectionalSequenceLstmWorkload()
338 …InitializeArmComputeTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeight… in NeonUnidirectionalSequenceLstmWorkload()
861 FreeTensorIfUnused(m_RecurrentToForgetWeightsTensor); in FreeUnusedTensors()
H A DNeonQuantizedLstmWorkload.hpp33 std::unique_ptr<arm_compute::Tensor> m_RecurrentToForgetWeightsTensor; member in armnn::NeonQuantizedLstmWorkload
H A DNeonQLstmWorkload.hpp34 std::unique_ptr<arm_compute::Tensor> m_RecurrentToForgetWeightsTensor; member in armnn::NeonQLstmWorkload
H A DNeonLstmFloatWorkload.hpp37 std::unique_ptr<arm_compute::Tensor> m_RecurrentToForgetWeightsTensor; member in armnn::NeonLstmFloatWorkload
H A DNeonUnidirectionalSequenceLstmWorkload.hpp48 std::unique_ptr<arm_compute::Tensor> m_RecurrentToForgetWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmWorkload
H A DNeonUnidirectionalSequenceLstmFloatWorkload.hpp50 std::unique_ptr<arm_compute::Tensor> m_RecurrentToForgetWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmFloatWorkload
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClQuantizedLstmWorkload.cpp84 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQuantizedLstmWorkload()
85 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in ClQuantizedLstmWorkload()
117 … m_RecurrentToInputWeightsTensor.get(), m_RecurrentToForgetWeightsTensor.get(), in ClQuantizedLstmWorkload()
130 …InitializeArmComputeClTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeig… in ClQuantizedLstmWorkload()
155 FreeTensorIfUnused(m_RecurrentToForgetWeightsTensor); in FreeUnusedTensors()
H A DClQLstmWorkload.cpp40 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload()
41 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in ClQLstmWorkload()
169 m_RecurrentToForgetWeightsTensor.get(), in ClQLstmWorkload()
188 …InitializeArmComputeClTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeig… in ClQLstmWorkload()
405 FreeTensorIfUnused(m_RecurrentToForgetWeightsTensor); in FreeUnusedTensors()
H A DClLstmFloatWorkload.cpp46 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload()
47 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in ClLstmFloatWorkload()
174m_RecurrentToForgetWeightsTensor.get(), m_RecurrentToCellWeightsTensor.get(), in ClLstmFloatWorkload()
187 …InitializeArmComputeClTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeig… in ClLstmFloatWorkload()
377 FreeTensorIfUnused(m_RecurrentToForgetWeightsTensor); in FreeUnusedTensors()
H A DClUnidirectionalSequenceLstmFloatWorkload.cpp162 m_RecurrentToForgetWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClUnidirectionalSequenceLstmFloatWorkload()
163 …BuildArmComputeTensor(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeights->GetTen… in ClUnidirectionalSequenceLstmFloatWorkload()
329 m_RecurrentToForgetWeightsTensor.get(), in ClUnidirectionalSequenceLstmFloatWorkload()
354 …InitializeArmComputeClTensorData(*m_RecurrentToForgetWeightsTensor, m_Data.m_RecurrentToForgetWeig… in ClUnidirectionalSequenceLstmFloatWorkload()
884 FreeTensorIfUnused(m_RecurrentToForgetWeightsTensor); in FreeUnusedTensors()
H A DClQuantizedLstmWorkload.hpp40 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToForgetWeightsTensor; member in armnn::ClQuantizedLstmWorkload
H A DClQLstmWorkload.hpp36 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToForgetWeightsTensor; member in armnn::ClQLstmWorkload
H A DClLstmFloatWorkload.hpp39 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToForgetWeightsTensor; member in armnn::ClLstmFloatWorkload
H A DClUnidirectionalSequenceLstmFloatWorkload.hpp52 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToForgetWeightsTensor; member in armnn::ClUnidirectionalSequenceLstmFloatWorkload
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefLstmWorkload.cpp24 …, m_RecurrentToForgetWeightsTensor(AssignScopedTensorHandle(descriptor.m_RecurrentToForgetWeights)) in RefLstmWorkload()
129m_RecurrentToForgetWeightsTensor->GetTensorInfo(), m_RecurrentToForgetWeightsTensor->GetConstTenso… in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.cpp28 …, m_RecurrentToForgetWeightsTensor(AssignScopedTensorHandle(descriptor.m_RecurrentToForgetWeights)) in RefUnidirectionalSequenceLstmWorkload()
160m_RecurrentToForgetWeightsTensor->GetTensorInfo(), m_RecurrentToForgetWeightsTensor->GetConstTenso… in Execute()
H A DRefQLstmWorkload.cpp24 …, m_RecurrentToForgetWeightsTensor(AssignScopedTensorHandle(descriptor.m_RecurrentToForgetWeights)) in RefQLstmWorkload()
123 m_RecurrentToForgetWeightsTensor->GetTensorInfo(), in Execute()
124 m_RecurrentToForgetWeightsTensor->GetConstTensor<void>()); in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.hpp36 std::unique_ptr<ScopedTensorHandle> m_RecurrentToForgetWeightsTensor; member in armnn::RefUnidirectionalSequenceLstmWorkload
H A DRefLstmWorkload.hpp31 std::unique_ptr<ScopedTensorHandle> m_RecurrentToForgetWeightsTensor; member in armnn::RefLstmWorkload
H A DRefQLstmWorkload.hpp32 std::unique_ptr<ScopedTensorHandle> m_RecurrentToForgetWeightsTensor; member in armnn::RefQLstmWorkload