Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonQuantizedLstmWorkload.cpp34 m_RecurrentToInputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQuantizedLstmWorkload()
35 …BuildArmComputeTensor(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeights->GetTenso… in NeonQuantizedLstmWorkload()
70 m_RecurrentToInputWeightsTensor.get(), in NeonQuantizedLstmWorkload()
95 InitializeArmComputeTensorData(*m_RecurrentToInputWeightsTensor, in NeonQuantizedLstmWorkload()
198 FreeTensorIfUnused(m_RecurrentToInputWeightsTensor); in FreeUnusedTensors()
H A DNeonQLstmWorkload.cpp126 m_RecurrentToInputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload()
127 …BuildArmComputeTensor(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeights->GetTenso… in NeonQLstmWorkload()
135 m_RecurrentToInputWeightsTensor.get(), in NeonQLstmWorkload()
194 …InitializeArmComputeTensorData(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeights); in NeonQLstmWorkload()
400 FreeTensorIfUnused(m_RecurrentToInputWeightsTensor); in FreeUnusedTensors()
H A DNeonLstmFloatWorkload.cpp65 m_RecurrentToInputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload()
66 …BuildArmComputeTensor(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeights->GetTenso… in NeonLstmFloatWorkload()
78 m_RecurrentToInputWeightsTensor.get(), in NeonLstmFloatWorkload()
200 InitializeArmComputeTensorData(*m_RecurrentToInputWeightsTensor, in NeonLstmFloatWorkload()
397 FreeTensorIfUnused(m_RecurrentToInputWeightsTensor); in FreeUnusedTensors()
H A DNeonUnidirectionalSequenceLstmFloatWorkload.cpp184 m_RecurrentToInputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmFloatWorkload()
185 …BuildArmComputeTensor(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeights->GetTenso… in NeonUnidirectionalSequenceLstmFloatWorkload()
197 m_RecurrentToInputWeightsTensor.get(), in NeonUnidirectionalSequenceLstmFloatWorkload()
367 …InitializeArmComputeTensorData(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeights); in NeonUnidirectionalSequenceLstmFloatWorkload()
887 FreeTensorIfUnused(m_RecurrentToInputWeightsTensor); in FreeUnusedTensors()
H A DNeonUnidirectionalSequenceLstmWorkload.cpp197 m_RecurrentToInputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmWorkload()
198 …BuildArmComputeTensor(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeights->GetTenso… in NeonUnidirectionalSequenceLstmWorkload()
209 m_RecurrentToInputWeightsTensor.get(), in NeonUnidirectionalSequenceLstmWorkload()
348 …InitializeArmComputeTensorData(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeights); in NeonUnidirectionalSequenceLstmWorkload()
860 FreeTensorIfUnused(m_RecurrentToInputWeightsTensor); in FreeUnusedTensors()
H A DNeonQuantizedLstmWorkload.hpp32 std::unique_ptr<arm_compute::Tensor> m_RecurrentToInputWeightsTensor; member in armnn::NeonQuantizedLstmWorkload
H A DNeonQLstmWorkload.hpp33 std::unique_ptr<arm_compute::Tensor> m_RecurrentToInputWeightsTensor; member in armnn::NeonQLstmWorkload
H A DNeonLstmFloatWorkload.hpp36 std::unique_ptr<arm_compute::Tensor> m_RecurrentToInputWeightsTensor; member in armnn::NeonLstmFloatWorkload
H A DNeonUnidirectionalSequenceLstmWorkload.hpp47 std::unique_ptr<arm_compute::Tensor> m_RecurrentToInputWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmWorkload
H A DNeonUnidirectionalSequenceLstmFloatWorkload.hpp49 std::unique_ptr<arm_compute::Tensor> m_RecurrentToInputWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmFloatWorkload
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClQuantizedLstmWorkload.cpp81 m_RecurrentToInputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQuantizedLstmWorkload()
82 …BuildArmComputeTensor(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeights->GetTenso… in ClQuantizedLstmWorkload()
117 … m_RecurrentToInputWeightsTensor.get(), m_RecurrentToForgetWeightsTensor.get(), in ClQuantizedLstmWorkload()
129 …InitializeArmComputeClTensorData(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeigh… in ClQuantizedLstmWorkload()
154 FreeTensorIfUnused(m_RecurrentToInputWeightsTensor); in FreeUnusedTensors()
H A DClQLstmWorkload.cpp128 m_RecurrentToInputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload()
129 …BuildArmComputeTensor(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeights->GetTenso… in ClQLstmWorkload()
137 m_RecurrentToInputWeightsTensor.get(), in ClQLstmWorkload()
200 …InitializeArmComputeClTensorData(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeight… in ClQLstmWorkload()
404 FreeTensorIfUnused(m_RecurrentToInputWeightsTensor); in FreeUnusedTensors()
H A DClLstmFloatWorkload.cpp70 m_RecurrentToInputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload()
71 …BuildArmComputeTensor(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeights->GetTenso… in ClLstmFloatWorkload()
83 m_RecurrentToInputWeightsTensor.get(), in ClLstmFloatWorkload()
197 …InitializeArmComputeClTensorData(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeight… in ClLstmFloatWorkload()
376 FreeTensorIfUnused(m_RecurrentToInputWeightsTensor); in FreeUnusedTensors()
H A DClUnidirectionalSequenceLstmFloatWorkload.cpp186 m_RecurrentToInputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClUnidirectionalSequenceLstmFloatWorkload()
187 …BuildArmComputeTensor(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeights->GetTenso… in ClUnidirectionalSequenceLstmFloatWorkload()
199 m_RecurrentToInputWeightsTensor.get(), in ClUnidirectionalSequenceLstmFloatWorkload()
364 …InitializeArmComputeClTensorData(*m_RecurrentToInputWeightsTensor, m_Data.m_RecurrentToInputWeight… in ClUnidirectionalSequenceLstmFloatWorkload()
883 FreeTensorIfUnused(m_RecurrentToInputWeightsTensor); in FreeUnusedTensors()
H A DClQuantizedLstmWorkload.hpp39 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToInputWeightsTensor; member in armnn::ClQuantizedLstmWorkload
H A DClQLstmWorkload.hpp35 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToInputWeightsTensor; member in armnn::ClQLstmWorkload
H A DClLstmFloatWorkload.hpp38 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToInputWeightsTensor; member in armnn::ClLstmFloatWorkload
H A DClUnidirectionalSequenceLstmFloatWorkload.hpp51 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToInputWeightsTensor; member in armnn::ClUnidirectionalSequenceLstmFloatWorkload
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefLstmWorkload.cpp23 … , m_RecurrentToInputWeightsTensor (AssignScopedTensorHandle(descriptor.m_RecurrentToInputWeights)) in RefLstmWorkload()
180 …m_RecurrentToInputWeightsTensor->GetTensorInfo(), m_RecurrentToInputWeightsTensor->GetConstTensor<… in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.cpp27 … , m_RecurrentToInputWeightsTensor (AssignScopedTensorHandle(descriptor.m_RecurrentToInputWeights)) in RefUnidirectionalSequenceLstmWorkload()
208 …m_RecurrentToInputWeightsTensor->GetTensorInfo(), m_RecurrentToInputWeightsTensor->GetConstTensor<… in Execute()
H A DRefQLstmWorkload.cpp23 … , m_RecurrentToInputWeightsTensor (AssignScopedTensorHandle(descriptor.m_RecurrentToInputWeights)) in RefQLstmWorkload()
216 …recurrentToInputWeightsDecoder = MakeDecoder<float>(m_RecurrentToInputWeightsTensor->GetTensorInfo… in Execute()
217 … m_RecurrentToInputWeightsTensor->GetConstTensor<void>()); in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.hpp35 std::unique_ptr<ScopedTensorHandle> m_RecurrentToInputWeightsTensor; member in armnn::RefUnidirectionalSequenceLstmWorkload
H A DRefLstmWorkload.hpp30 std::unique_ptr<ScopedTensorHandle> m_RecurrentToInputWeightsTensor; member in armnn::RefLstmWorkload
H A DRefQLstmWorkload.hpp31 std::unique_ptr<ScopedTensorHandle> m_RecurrentToInputWeightsTensor; member in armnn::RefQLstmWorkload