Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonQuantizedLstmWorkload.cpp25 m_InputToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQuantizedLstmWorkload()
26 …BuildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo(… in NeonQuantizedLstmWorkload()
67 m_InputToForgetWeightsTensor.get(), in NeonQuantizedLstmWorkload()
86 InitializeArmComputeTensorData(*m_InputToForgetWeightsTensor, in NeonQuantizedLstmWorkload()
195 FreeTensorIfUnused(m_InputToForgetWeightsTensor); in FreeUnusedTensors()
H A DNeonQLstmWorkload.cpp29 m_InputToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload()
30 …BuildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo(… in NeonQLstmWorkload()
161 m_InputToForgetWeightsTensor.get(), in NeonQLstmWorkload()
178 InitializeArmComputeTensorData(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights); in NeonQLstmWorkload()
396 FreeTensorIfUnused(m_InputToForgetWeightsTensor); in FreeUnusedTensors()
H A DNeonLstmFloatWorkload.cpp32 m_InputToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload()
33 …BuildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo(… in NeonLstmFloatWorkload()
167 …m_LstmLayer.configure(&input, m_InputToForgetWeightsTensor.get(), m_InputToCellWeightsTensor.get(), in NeonLstmFloatWorkload()
177 InitializeArmComputeTensorData(*m_InputToForgetWeightsTensor, in NeonLstmFloatWorkload()
394 FreeTensorIfUnused(m_InputToForgetWeightsTensor); in FreeUnusedTensors()
H A DNeonUnidirectionalSequenceLstmFloatWorkload.cpp151 m_InputToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmFloatWorkload()
152 …BuildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo(… in NeonUnidirectionalSequenceLstmFloatWorkload()
329 m_InputToForgetWeightsTensor.get(), in NeonUnidirectionalSequenceLstmFloatWorkload()
354 InitializeArmComputeTensorData(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights); in NeonUnidirectionalSequenceLstmFloatWorkload()
884 FreeTensorIfUnused(m_InputToForgetWeightsTensor); in FreeUnusedTensors()
H A DNeonUnidirectionalSequenceLstmWorkload.cpp164 m_InputToForgetWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmWorkload()
165 …BuildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo(… in NeonUnidirectionalSequenceLstmWorkload()
316 m_InputToForgetWeightsTensor.get(), in NeonUnidirectionalSequenceLstmWorkload()
335 InitializeArmComputeTensorData(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights); in NeonUnidirectionalSequenceLstmWorkload()
857 FreeTensorIfUnused(m_InputToForgetWeightsTensor); in FreeUnusedTensors()
H A DNeonQuantizedLstmWorkload.hpp29 std::unique_ptr<arm_compute::Tensor> m_InputToForgetWeightsTensor; member in armnn::NeonQuantizedLstmWorkload
H A DNeonQLstmWorkload.hpp29 std::unique_ptr<arm_compute::Tensor> m_InputToForgetWeightsTensor; member in armnn::NeonQLstmWorkload
H A DNeonLstmFloatWorkload.hpp33 std::unique_ptr<arm_compute::Tensor> m_InputToForgetWeightsTensor; member in armnn::NeonLstmFloatWorkload
H A DNeonUnidirectionalSequenceLstmWorkload.hpp44 std::unique_ptr<arm_compute::Tensor> m_InputToForgetWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmWorkload
H A DNeonUnidirectionalSequenceLstmFloatWorkload.hpp46 std::unique_ptr<arm_compute::Tensor> m_InputToForgetWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmFloatWorkload
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClQuantizedLstmWorkload.cpp72 m_InputToForgetWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQuantizedLstmWorkload()
73 …BuildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo(… in ClQuantizedLstmWorkload()
115 m_InputToForgetWeightsTensor.get(), in ClQuantizedLstmWorkload()
126 …InitializeArmComputeClTensorData(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights); in ClQuantizedLstmWorkload()
151 FreeTensorIfUnused(m_InputToForgetWeightsTensor); in FreeUnusedTensors()
H A DClQLstmWorkload.cpp31 m_InputToForgetWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload()
32 …BuildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo(… in ClQLstmWorkload()
166 m_InputToForgetWeightsTensor.get(), in ClQLstmWorkload()
184 InitializeArmComputeClTensorData(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights); in ClQLstmWorkload()
400 FreeTensorIfUnused(m_InputToForgetWeightsTensor); in FreeUnusedTensors()
H A DClLstmFloatWorkload.cpp37 m_InputToForgetWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload()
38 …BuildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo(… in ClLstmFloatWorkload()
172 m_LstmLayer.configure(clCompileContext, &input, m_InputToForgetWeightsTensor.get(), in ClLstmFloatWorkload()
184 …InitializeArmComputeClTensorData(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights); in ClLstmFloatWorkload()
373 FreeTensorIfUnused(m_InputToForgetWeightsTensor); in FreeUnusedTensors()
H A DClUnidirectionalSequenceLstmFloatWorkload.cpp153 m_InputToForgetWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClUnidirectionalSequenceLstmFloatWorkload()
154 …BuildArmComputeTensor(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights->GetTensorInfo(… in ClUnidirectionalSequenceLstmFloatWorkload()
326 m_InputToForgetWeightsTensor.get(), in ClUnidirectionalSequenceLstmFloatWorkload()
351 InitializeArmComputeClTensorData(*m_InputToForgetWeightsTensor, m_Data.m_InputToForgetWeights); in ClUnidirectionalSequenceLstmFloatWorkload()
880 FreeTensorIfUnused(m_InputToForgetWeightsTensor); in FreeUnusedTensors()
H A DClQuantizedLstmWorkload.hpp36 std::unique_ptr<arm_compute::CLTensor> m_InputToForgetWeightsTensor; member in armnn::ClQuantizedLstmWorkload
H A DClQLstmWorkload.hpp31 std::unique_ptr<arm_compute::CLTensor> m_InputToForgetWeightsTensor; member in armnn::ClQLstmWorkload
H A DClLstmFloatWorkload.hpp35 std::unique_ptr<arm_compute::CLTensor> m_InputToForgetWeightsTensor; member in armnn::ClLstmFloatWorkload
H A DClUnidirectionalSequenceLstmFloatWorkload.hpp48 std::unique_ptr<arm_compute::CLTensor> m_InputToForgetWeightsTensor; member in armnn::ClUnidirectionalSequenceLstmFloatWorkload
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefLstmWorkload.cpp20 , m_InputToForgetWeightsTensor (AssignScopedTensorHandle(descriptor.m_InputToForgetWeights)) in RefLstmWorkload()
121 …m_InputToForgetWeightsTensor->GetTensorInfo(), m_InputToForgetWeightsTensor->GetConstTensor<void>(… in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.cpp24 , m_InputToForgetWeightsTensor (AssignScopedTensorHandle(descriptor.m_InputToForgetWeights)) in RefUnidirectionalSequenceLstmWorkload()
152 …m_InputToForgetWeightsTensor->GetTensorInfo(), m_InputToForgetWeightsTensor->GetConstTensor<void>(… in Execute()
H A DRefQLstmWorkload.cpp19 … , m_InputToForgetWeightsTensor (AssignScopedTensorHandle(descriptor.m_InputToForgetWeights)) in RefQLstmWorkload()
116 …m_InputToForgetWeightsTensor->GetTensorInfo(), m_InputToForgetWeightsTensor->GetConstTensor<void>(… in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.hpp32 std::unique_ptr<ScopedTensorHandle> m_InputToForgetWeightsTensor; member in armnn::RefUnidirectionalSequenceLstmWorkload
H A DRefLstmWorkload.hpp27 std::unique_ptr<ScopedTensorHandle> m_InputToForgetWeightsTensor; member in armnn::RefLstmWorkload
H A DRefQLstmWorkload.hpp27 std::unique_ptr<ScopedTensorHandle> m_InputToForgetWeightsTensor; member in armnn::RefQLstmWorkload