Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonQuantizedLstmWorkload.cpp43 m_RecurrentToOutputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQuantizedLstmWorkload()
44 …BuildArmComputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTen… in NeonQuantizedLstmWorkload()
73 m_RecurrentToOutputWeightsTensor.get(), in NeonQuantizedLstmWorkload()
104 InitializeArmComputeTensorData(*m_RecurrentToOutputWeightsTensor, in NeonQuantizedLstmWorkload()
201 FreeTensorIfUnused(m_RecurrentToOutputWeightsTensor); in FreeUnusedTensors()
H A DNeonQLstmWorkload.cpp44 m_RecurrentToOutputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload()
45 …BuildArmComputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTen… in NeonQLstmWorkload()
166 m_RecurrentToOutputWeightsTensor.get(), in NeonQLstmWorkload()
184 …InitializeArmComputeTensorData(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeight… in NeonQLstmWorkload()
403 FreeTensorIfUnused(m_RecurrentToOutputWeightsTensor); in FreeUnusedTensors()
H A DNeonLstmFloatWorkload.cpp47 m_RecurrentToOutputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload()
48 …BuildArmComputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTen… in NeonLstmFloatWorkload()
169 … m_RecurrentToCellWeightsTensor.get(), m_RecurrentToOutputWeightsTensor.get(), in NeonLstmFloatWorkload()
187 InitializeArmComputeTensorData(*m_RecurrentToOutputWeightsTensor, in NeonLstmFloatWorkload()
400 FreeTensorIfUnused(m_RecurrentToOutputWeightsTensor); in FreeUnusedTensors()
H A DNeonUnidirectionalSequenceLstmFloatWorkload.cpp166 m_RecurrentToOutputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmFloatWorkload()
167 …BuildArmComputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTen… in NeonUnidirectionalSequenceLstmFloatWorkload()
334 m_RecurrentToOutputWeightsTensor.get(), in NeonUnidirectionalSequenceLstmFloatWorkload()
359 …InitializeArmComputeTensorData(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeight… in NeonUnidirectionalSequenceLstmFloatWorkload()
890 FreeTensorIfUnused(m_RecurrentToOutputWeightsTensor); in FreeUnusedTensors()
H A DNeonUnidirectionalSequenceLstmWorkload.cpp179 m_RecurrentToOutputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmWorkload()
180 …BuildArmComputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTen… in NeonUnidirectionalSequenceLstmWorkload()
321 m_RecurrentToOutputWeightsTensor.get(), in NeonUnidirectionalSequenceLstmWorkload()
340 …InitializeArmComputeTensorData(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeight… in NeonUnidirectionalSequenceLstmWorkload()
863 FreeTensorIfUnused(m_RecurrentToOutputWeightsTensor); in FreeUnusedTensors()
H A DNeonQuantizedLstmWorkload.hpp35 std::unique_ptr<arm_compute::Tensor> m_RecurrentToOutputWeightsTensor; member in armnn::NeonQuantizedLstmWorkload
H A DNeonQLstmWorkload.hpp36 std::unique_ptr<arm_compute::Tensor> m_RecurrentToOutputWeightsTensor; member in armnn::NeonQLstmWorkload
H A DNeonLstmFloatWorkload.hpp39 std::unique_ptr<arm_compute::Tensor> m_RecurrentToOutputWeightsTensor; member in armnn::NeonLstmFloatWorkload
H A DNeonUnidirectionalSequenceLstmWorkload.hpp50 std::unique_ptr<arm_compute::Tensor> m_RecurrentToOutputWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmWorkload
H A DNeonUnidirectionalSequenceLstmFloatWorkload.hpp52 std::unique_ptr<arm_compute::Tensor> m_RecurrentToOutputWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmFloatWorkload
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClQuantizedLstmWorkload.cpp90 m_RecurrentToOutputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQuantizedLstmWorkload()
91 …BuildArmComputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTen… in ClQuantizedLstmWorkload()
118 … m_RecurrentToCellWeightsTensor.get(), m_RecurrentToOutputWeightsTensor.get(), in ClQuantizedLstmWorkload()
132 …InitializeArmComputeClTensorData(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeig… in ClQuantizedLstmWorkload()
157 FreeTensorIfUnused(m_RecurrentToOutputWeightsTensor); in FreeUnusedTensors()
H A DClQLstmWorkload.cpp46 m_RecurrentToOutputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload()
47 …BuildArmComputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTen… in ClQLstmWorkload()
171 m_RecurrentToOutputWeightsTensor.get(), in ClQLstmWorkload()
190 …InitializeArmComputeClTensorData(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeig… in ClQLstmWorkload()
407 FreeTensorIfUnused(m_RecurrentToOutputWeightsTensor); in FreeUnusedTensors()
H A DClLstmFloatWorkload.cpp52 m_RecurrentToOutputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload()
53 …BuildArmComputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTen… in ClLstmFloatWorkload()
175 m_RecurrentToOutputWeightsTensor.get(), m_ForgetGateBiasTensor.get(), in ClLstmFloatWorkload()
189 …InitializeArmComputeClTensorData(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeig… in ClLstmFloatWorkload()
379 FreeTensorIfUnused(m_RecurrentToOutputWeightsTensor); in FreeUnusedTensors()
H A DClUnidirectionalSequenceLstmFloatWorkload.cpp168 m_RecurrentToOutputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClUnidirectionalSequenceLstmFloatWorkload()
169 …BuildArmComputeTensor(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeights->GetTen… in ClUnidirectionalSequenceLstmFloatWorkload()
331 m_RecurrentToOutputWeightsTensor.get(), in ClUnidirectionalSequenceLstmFloatWorkload()
356 …InitializeArmComputeClTensorData(*m_RecurrentToOutputWeightsTensor, m_Data.m_RecurrentToOutputWeig… in ClUnidirectionalSequenceLstmFloatWorkload()
886 FreeTensorIfUnused(m_RecurrentToOutputWeightsTensor); in FreeUnusedTensors()
H A DClQuantizedLstmWorkload.hpp42 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToOutputWeightsTensor; member in armnn::ClQuantizedLstmWorkload
H A DClQLstmWorkload.hpp38 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToOutputWeightsTensor; member in armnn::ClQLstmWorkload
H A DClLstmFloatWorkload.hpp41 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToOutputWeightsTensor; member in armnn::ClLstmFloatWorkload
H A DClUnidirectionalSequenceLstmFloatWorkload.hpp54 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToOutputWeightsTensor; member in armnn::ClUnidirectionalSequenceLstmFloatWorkload
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefLstmWorkload.cpp26 …, m_RecurrentToOutputWeightsTensor(AssignScopedTensorHandle(descriptor.m_RecurrentToOutputWeights)) in RefLstmWorkload()
133 …m_RecurrentToOutputWeightsTensor->GetTensorInfo(), m_RecurrentToOutputWeightsTensor->GetConstTenso… in Execute()
156 const TensorShape& recurrentToOutputWeightsShape = m_RecurrentToOutputWeightsTensor->GetShape(); in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.cpp30 …, m_RecurrentToOutputWeightsTensor(AssignScopedTensorHandle(descriptor.m_RecurrentToOutputWeights)) in RefUnidirectionalSequenceLstmWorkload()
136 const TensorShape& recurrentToOutputWeightsShape = m_RecurrentToOutputWeightsTensor->GetShape(); in Execute()
164 …m_RecurrentToOutputWeightsTensor->GetTensorInfo(), m_RecurrentToOutputWeightsTensor->GetConstTenso… in Execute()
H A DRefQLstmWorkload.cpp26 …, m_RecurrentToOutputWeightsTensor(AssignScopedTensorHandle(descriptor.m_RecurrentToOutputWeights)) in RefQLstmWorkload()
128 m_RecurrentToOutputWeightsTensor->GetTensorInfo(), in Execute()
129 m_RecurrentToOutputWeightsTensor->GetConstTensor<void>()); in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.hpp38 std::unique_ptr<ScopedTensorHandle> m_RecurrentToOutputWeightsTensor; member in armnn::RefUnidirectionalSequenceLstmWorkload
H A DRefLstmWorkload.hpp33 std::unique_ptr<ScopedTensorHandle> m_RecurrentToOutputWeightsTensor; member in armnn::RefLstmWorkload
H A DRefQLstmWorkload.hpp34 std::unique_ptr<ScopedTensorHandle> m_RecurrentToOutputWeightsTensor; member in armnn::RefQLstmWorkload