Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonQuantizedLstmWorkload.cpp40 m_RecurrentToCellWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQuantizedLstmWorkload()
41 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in NeonQuantizedLstmWorkload()
72 m_RecurrentToCellWeightsTensor.get(), in NeonQuantizedLstmWorkload()
101 InitializeArmComputeTensorData(*m_RecurrentToCellWeightsTensor, in NeonQuantizedLstmWorkload()
200 FreeTensorIfUnused(m_RecurrentToCellWeightsTensor); in FreeUnusedTensors()
H A DNeonQLstmWorkload.cpp41 m_RecurrentToCellWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload()
42 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in NeonQLstmWorkload()
165 m_RecurrentToCellWeightsTensor.get(), in NeonQLstmWorkload()
183 …InitializeArmComputeTensorData(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights); in NeonQLstmWorkload()
402 FreeTensorIfUnused(m_RecurrentToCellWeightsTensor); in FreeUnusedTensors()
H A DNeonLstmFloatWorkload.cpp44 m_RecurrentToCellWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload()
45 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in NeonLstmFloatWorkload()
169 … m_RecurrentToCellWeightsTensor.get(), m_RecurrentToOutputWeightsTensor.get(), in NeonLstmFloatWorkload()
185 InitializeArmComputeTensorData(*m_RecurrentToCellWeightsTensor, in NeonLstmFloatWorkload()
399 FreeTensorIfUnused(m_RecurrentToCellWeightsTensor); in FreeUnusedTensors()
H A DNeonUnidirectionalSequenceLstmFloatWorkload.cpp163 m_RecurrentToCellWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmFloatWorkload()
164 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in NeonUnidirectionalSequenceLstmFloatWorkload()
333 m_RecurrentToCellWeightsTensor.get(), in NeonUnidirectionalSequenceLstmFloatWorkload()
358 … InitializeArmComputeTensorData(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights); in NeonUnidirectionalSequenceLstmFloatWorkload()
889 FreeTensorIfUnused(m_RecurrentToCellWeightsTensor); in FreeUnusedTensors()
H A DNeonUnidirectionalSequenceLstmWorkload.cpp176 m_RecurrentToCellWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmWorkload()
177 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in NeonUnidirectionalSequenceLstmWorkload()
320 m_RecurrentToCellWeightsTensor.get(), in NeonUnidirectionalSequenceLstmWorkload()
339 … InitializeArmComputeTensorData(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights); in NeonUnidirectionalSequenceLstmWorkload()
862 FreeTensorIfUnused(m_RecurrentToCellWeightsTensor); in FreeUnusedTensors()
H A DNeonQuantizedLstmWorkload.hpp34 std::unique_ptr<arm_compute::Tensor> m_RecurrentToCellWeightsTensor; member in armnn::NeonQuantizedLstmWorkload
H A DNeonQLstmWorkload.hpp35 std::unique_ptr<arm_compute::Tensor> m_RecurrentToCellWeightsTensor; member in armnn::NeonQLstmWorkload
H A DNeonLstmFloatWorkload.hpp38 std::unique_ptr<arm_compute::Tensor> m_RecurrentToCellWeightsTensor; member in armnn::NeonLstmFloatWorkload
H A DNeonUnidirectionalSequenceLstmWorkload.hpp49 std::unique_ptr<arm_compute::Tensor> m_RecurrentToCellWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmWorkload
H A DNeonUnidirectionalSequenceLstmFloatWorkload.hpp51 std::unique_ptr<arm_compute::Tensor> m_RecurrentToCellWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmFloatWorkload
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClQuantizedLstmWorkload.cpp87 m_RecurrentToCellWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQuantizedLstmWorkload()
88 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in ClQuantizedLstmWorkload()
118 … m_RecurrentToCellWeightsTensor.get(), m_RecurrentToOutputWeightsTensor.get(), in ClQuantizedLstmWorkload()
131 …InitializeArmComputeClTensorData(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeight… in ClQuantizedLstmWorkload()
156 FreeTensorIfUnused(m_RecurrentToCellWeightsTensor); in FreeUnusedTensors()
H A DClQLstmWorkload.cpp43 m_RecurrentToCellWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload()
44 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in ClQLstmWorkload()
170 m_RecurrentToCellWeightsTensor.get(), in ClQLstmWorkload()
189 …InitializeArmComputeClTensorData(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeight… in ClQLstmWorkload()
406 FreeTensorIfUnused(m_RecurrentToCellWeightsTensor); in FreeUnusedTensors()
H A DClLstmFloatWorkload.cpp49 m_RecurrentToCellWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload()
50 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in ClLstmFloatWorkload()
174 … m_RecurrentToForgetWeightsTensor.get(), m_RecurrentToCellWeightsTensor.get(), in ClLstmFloatWorkload()
188 …InitializeArmComputeClTensorData(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeight… in ClLstmFloatWorkload()
378 FreeTensorIfUnused(m_RecurrentToCellWeightsTensor); in FreeUnusedTensors()
H A DClUnidirectionalSequenceLstmFloatWorkload.cpp165 m_RecurrentToCellWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClUnidirectionalSequenceLstmFloatWorkload()
166 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in ClUnidirectionalSequenceLstmFloatWorkload()
330 m_RecurrentToCellWeightsTensor.get(), in ClUnidirectionalSequenceLstmFloatWorkload()
355 …InitializeArmComputeClTensorData(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights); in ClUnidirectionalSequenceLstmFloatWorkload()
885 FreeTensorIfUnused(m_RecurrentToCellWeightsTensor); in FreeUnusedTensors()
H A DClQuantizedLstmWorkload.hpp41 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToCellWeightsTensor; member in armnn::ClQuantizedLstmWorkload
H A DClQLstmWorkload.hpp37 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToCellWeightsTensor; member in armnn::ClQLstmWorkload
H A DClLstmFloatWorkload.hpp40 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToCellWeightsTensor; member in armnn::ClLstmFloatWorkload
H A DClUnidirectionalSequenceLstmFloatWorkload.hpp53 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToCellWeightsTensor; member in armnn::ClUnidirectionalSequenceLstmFloatWorkload
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefLstmWorkload.cpp25 … , m_RecurrentToCellWeightsTensor (AssignScopedTensorHandle(descriptor.m_RecurrentToCellWeights)) in RefLstmWorkload()
131 …m_RecurrentToCellWeightsTensor->GetTensorInfo(), m_RecurrentToCellWeightsTensor->GetConstTensor<vo… in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.cpp29 … , m_RecurrentToCellWeightsTensor (AssignScopedTensorHandle(descriptor.m_RecurrentToCellWeights)) in RefUnidirectionalSequenceLstmWorkload()
162 …m_RecurrentToCellWeightsTensor->GetTensorInfo(), m_RecurrentToCellWeightsTensor->GetConstTensor<vo… in Execute()
H A DRefQLstmWorkload.cpp25 … , m_RecurrentToCellWeightsTensor (AssignScopedTensorHandle(descriptor.m_RecurrentToCellWeights)) in RefQLstmWorkload()
126 …m_RecurrentToCellWeightsTensor->GetTensorInfo(), m_RecurrentToCellWeightsTensor->GetConstTensor<vo… in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.hpp37 std::unique_ptr<ScopedTensorHandle> m_RecurrentToCellWeightsTensor; member in armnn::RefUnidirectionalSequenceLstmWorkload
H A DRefLstmWorkload.hpp32 std::unique_ptr<ScopedTensorHandle> m_RecurrentToCellWeightsTensor; member in armnn::RefLstmWorkload
H A DRefQLstmWorkload.hpp33 std::unique_ptr<ScopedTensorHandle> m_RecurrentToCellWeightsTensor; member in armnn::RefQLstmWorkload