Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonQuantizedLstmWorkload.cpp31 m_InputToOutputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQuantizedLstmWorkload()
32 …BuildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo(… in NeonQuantizedLstmWorkload()
69 m_InputToOutputWeightsTensor.get(), in NeonQuantizedLstmWorkload()
92 InitializeArmComputeTensorData(*m_InputToOutputWeightsTensor, in NeonQuantizedLstmWorkload()
197 FreeTensorIfUnused(m_InputToOutputWeightsTensor); in FreeUnusedTensors()
H A DNeonQLstmWorkload.cpp35 m_InputToOutputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload()
36 …BuildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo(… in NeonQLstmWorkload()
163 m_InputToOutputWeightsTensor.get(), in NeonQLstmWorkload()
180 InitializeArmComputeTensorData(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights); in NeonQLstmWorkload()
398 FreeTensorIfUnused(m_InputToOutputWeightsTensor); in FreeUnusedTensors()
H A DNeonLstmFloatWorkload.cpp38 m_InputToOutputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload()
39 …BuildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo(… in NeonLstmFloatWorkload()
168 … m_InputToOutputWeightsTensor.get(), m_RecurrentToForgetWeightsTensor.get(), in NeonLstmFloatWorkload()
181 InitializeArmComputeTensorData(*m_InputToOutputWeightsTensor, in NeonLstmFloatWorkload()
396 FreeTensorIfUnused(m_InputToOutputWeightsTensor); in FreeUnusedTensors()
H A DNeonUnidirectionalSequenceLstmFloatWorkload.cpp157 m_InputToOutputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmFloatWorkload()
158 …BuildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo(… in NeonUnidirectionalSequenceLstmFloatWorkload()
331 m_InputToOutputWeightsTensor.get(), in NeonUnidirectionalSequenceLstmFloatWorkload()
356 InitializeArmComputeTensorData(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights); in NeonUnidirectionalSequenceLstmFloatWorkload()
886 FreeTensorIfUnused(m_InputToOutputWeightsTensor); in FreeUnusedTensors()
H A DNeonUnidirectionalSequenceLstmWorkload.cpp170 m_InputToOutputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmWorkload()
171 …BuildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo(… in NeonUnidirectionalSequenceLstmWorkload()
318 m_InputToOutputWeightsTensor.get(), in NeonUnidirectionalSequenceLstmWorkload()
337 InitializeArmComputeTensorData(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights); in NeonUnidirectionalSequenceLstmWorkload()
859 FreeTensorIfUnused(m_InputToOutputWeightsTensor); in FreeUnusedTensors()
H A DNeonQuantizedLstmWorkload.hpp31 std::unique_ptr<arm_compute::Tensor> m_InputToOutputWeightsTensor; member in armnn::NeonQuantizedLstmWorkload
H A DNeonQLstmWorkload.hpp31 std::unique_ptr<arm_compute::Tensor> m_InputToOutputWeightsTensor; member in armnn::NeonQLstmWorkload
H A DNeonLstmFloatWorkload.hpp35 std::unique_ptr<arm_compute::Tensor> m_InputToOutputWeightsTensor; member in armnn::NeonLstmFloatWorkload
H A DNeonUnidirectionalSequenceLstmWorkload.hpp46 std::unique_ptr<arm_compute::Tensor> m_InputToOutputWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmWorkload
H A DNeonUnidirectionalSequenceLstmFloatWorkload.hpp48 std::unique_ptr<arm_compute::Tensor> m_InputToOutputWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmFloatWorkload
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClQuantizedLstmWorkload.cpp78 m_InputToOutputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQuantizedLstmWorkload()
79 …BuildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo(… in ClQuantizedLstmWorkload()
116 … m_InputToCellWeightsTensor.get(), m_InputToOutputWeightsTensor.get(), in ClQuantizedLstmWorkload()
128 …InitializeArmComputeClTensorData(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights); in ClQuantizedLstmWorkload()
153 FreeTensorIfUnused(m_InputToOutputWeightsTensor); in FreeUnusedTensors()
H A DClQLstmWorkload.cpp37 m_InputToOutputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload()
38 …BuildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo(… in ClQLstmWorkload()
168 m_InputToOutputWeightsTensor.get(), in ClQLstmWorkload()
186 InitializeArmComputeClTensorData(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights); in ClQLstmWorkload()
402 FreeTensorIfUnused(m_InputToOutputWeightsTensor); in FreeUnusedTensors()
H A DClLstmFloatWorkload.cpp43 m_InputToOutputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload()
44 …BuildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo(… in ClLstmFloatWorkload()
173 m_InputToCellWeightsTensor.get(), m_InputToOutputWeightsTensor.get(), in ClLstmFloatWorkload()
186 …InitializeArmComputeClTensorData(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights); in ClLstmFloatWorkload()
375 FreeTensorIfUnused(m_InputToOutputWeightsTensor); in FreeUnusedTensors()
H A DClUnidirectionalSequenceLstmFloatWorkload.cpp159 m_InputToOutputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClUnidirectionalSequenceLstmFloatWorkload()
160 …BuildArmComputeTensor(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights->GetTensorInfo(… in ClUnidirectionalSequenceLstmFloatWorkload()
328 m_InputToOutputWeightsTensor.get(), in ClUnidirectionalSequenceLstmFloatWorkload()
353 InitializeArmComputeClTensorData(*m_InputToOutputWeightsTensor, m_Data.m_InputToOutputWeights); in ClUnidirectionalSequenceLstmFloatWorkload()
882 FreeTensorIfUnused(m_InputToOutputWeightsTensor); in FreeUnusedTensors()
H A DClQuantizedLstmWorkload.hpp38 std::unique_ptr<arm_compute::CLTensor> m_InputToOutputWeightsTensor; member in armnn::ClQuantizedLstmWorkload
H A DClQLstmWorkload.hpp33 std::unique_ptr<arm_compute::CLTensor> m_InputToOutputWeightsTensor; member in armnn::ClQLstmWorkload
H A DClLstmFloatWorkload.hpp37 std::unique_ptr<arm_compute::CLTensor> m_InputToOutputWeightsTensor; member in armnn::ClLstmFloatWorkload
H A DClUnidirectionalSequenceLstmFloatWorkload.hpp50 std::unique_ptr<arm_compute::CLTensor> m_InputToOutputWeightsTensor; member in armnn::ClUnidirectionalSequenceLstmFloatWorkload
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefLstmWorkload.cpp22 , m_InputToOutputWeightsTensor (AssignScopedTensorHandle(descriptor.m_InputToOutputWeights)) in RefLstmWorkload()
75 const uint32_t nCell = m_InputToOutputWeightsTensor->GetShape()[0]; in Execute()
125 …m_InputToOutputWeightsTensor->GetTensorInfo(), m_InputToOutputWeightsTensor->GetConstTensor<void>(… in Execute()
155 const TensorShape& inputToOutputWeightsShape = m_InputToOutputWeightsTensor->GetShape(); in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.cpp26 , m_InputToOutputWeightsTensor (AssignScopedTensorHandle(descriptor.m_InputToOutputWeights)) in RefUnidirectionalSequenceLstmWorkload()
135 const TensorShape& inputToOutputWeightsShape = m_InputToOutputWeightsTensor->GetShape(); in Execute()
156 …m_InputToOutputWeightsTensor->GetTensorInfo(), m_InputToOutputWeightsTensor->GetConstTensor<void>(… in Execute()
H A DRefQLstmWorkload.cpp21 … , m_InputToOutputWeightsTensor (AssignScopedTensorHandle(descriptor.m_InputToOutputWeights)) in RefQLstmWorkload()
120 …m_InputToOutputWeightsTensor->GetTensorInfo(), m_InputToOutputWeightsTensor->GetConstTensor<void>(… in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.hpp34 std::unique_ptr<ScopedTensorHandle> m_InputToOutputWeightsTensor; member in armnn::RefUnidirectionalSequenceLstmWorkload
H A DRefLstmWorkload.hpp29 std::unique_ptr<ScopedTensorHandle> m_InputToOutputWeightsTensor; member in armnn::RefLstmWorkload
H A DRefQLstmWorkload.hpp29 std::unique_ptr<ScopedTensorHandle> m_InputToOutputWeightsTensor; member in armnn::RefQLstmWorkload