Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonQuantizedLstmWorkload.cpp22 m_InputToInputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQuantizedLstmWorkload()
23 …BuildArmComputeTensor(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights->GetTensorInfo()); in NeonQuantizedLstmWorkload()
66 m_InputToInputWeightsTensor.get(), in NeonQuantizedLstmWorkload()
83 InitializeArmComputeTensorData(*m_InputToInputWeightsTensor, in NeonQuantizedLstmWorkload()
194 FreeTensorIfUnused(m_InputToInputWeightsTensor); in FreeUnusedTensors()
H A DNeonQLstmWorkload.cpp123 m_InputToInputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload()
124 …BuildArmComputeTensor(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights->GetTensorInfo()); in NeonQLstmWorkload()
134 m_InputToInputWeightsTensor.get(), in NeonQLstmWorkload()
193 … InitializeArmComputeTensorData(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights); in NeonQLstmWorkload()
395 FreeTensorIfUnused(m_InputToInputWeightsTensor); in FreeUnusedTensors()
H A DNeonLstmFloatWorkload.cpp62 m_InputToInputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload()
63 …BuildArmComputeTensor(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights->GetTensorInfo()); in NeonLstmFloatWorkload()
77 lstm_param.set_cifg_params(m_InputToInputWeightsTensor.get(), in NeonLstmFloatWorkload()
198 InitializeArmComputeTensorData(*m_InputToInputWeightsTensor, in NeonLstmFloatWorkload()
393 FreeTensorIfUnused(m_InputToInputWeightsTensor); in FreeUnusedTensors()
H A DNeonUnidirectionalSequenceLstmFloatWorkload.cpp181 m_InputToInputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmFloatWorkload()
182 …BuildArmComputeTensor(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights->GetTensorInfo()); in NeonUnidirectionalSequenceLstmFloatWorkload()
196 lstm_param.set_cifg_params(m_InputToInputWeightsTensor.get(), in NeonUnidirectionalSequenceLstmFloatWorkload()
366 InitializeArmComputeTensorData(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights); in NeonUnidirectionalSequenceLstmFloatWorkload()
883 FreeTensorIfUnused(m_InputToInputWeightsTensor); in FreeUnusedTensors()
H A DNeonUnidirectionalSequenceLstmWorkload.cpp194 m_InputToInputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonUnidirectionalSequenceLstmWorkload()
195 …BuildArmComputeTensor(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights->GetTensorInfo()); in NeonUnidirectionalSequenceLstmWorkload()
208 lstm_param.set_cifg_params(m_InputToInputWeightsTensor.get(), in NeonUnidirectionalSequenceLstmWorkload()
347 InitializeArmComputeTensorData(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights); in NeonUnidirectionalSequenceLstmWorkload()
856 FreeTensorIfUnused(m_InputToInputWeightsTensor); in FreeUnusedTensors()
H A DNeonQuantizedLstmWorkload.hpp28 std::unique_ptr<arm_compute::Tensor> m_InputToInputWeightsTensor; member in armnn::NeonQuantizedLstmWorkload
H A DNeonQLstmWorkload.hpp28 std::unique_ptr<arm_compute::Tensor> m_InputToInputWeightsTensor; member in armnn::NeonQLstmWorkload
H A DNeonLstmFloatWorkload.hpp32 std::unique_ptr<arm_compute::Tensor> m_InputToInputWeightsTensor; member in armnn::NeonLstmFloatWorkload
H A DNeonUnidirectionalSequenceLstmWorkload.hpp43 std::unique_ptr<arm_compute::Tensor> m_InputToInputWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmWorkload
H A DNeonUnidirectionalSequenceLstmFloatWorkload.hpp45 std::unique_ptr<arm_compute::Tensor> m_InputToInputWeightsTensor; member in armnn::NeonUnidirectionalSequenceLstmFloatWorkload
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClQuantizedLstmWorkload.cpp69 m_InputToInputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQuantizedLstmWorkload()
70 …BuildArmComputeTensor(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights->GetTensorInfo()); in ClQuantizedLstmWorkload()
114 … m_QuantizedLstmLayer.configure(clCompileContext, &inputTensor, m_InputToInputWeightsTensor.get(), in ClQuantizedLstmWorkload()
125 … InitializeArmComputeClTensorData(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights); in ClQuantizedLstmWorkload()
150 FreeTensorIfUnused(m_InputToInputWeightsTensor); in FreeUnusedTensors()
H A DClQLstmWorkload.cpp125 m_InputToInputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload()
126 …BuildArmComputeTensor(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights->GetTensorInfo()); in ClQLstmWorkload()
136 m_InputToInputWeightsTensor.get(), in ClQLstmWorkload()
199 … InitializeArmComputeClTensorData(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights); in ClQLstmWorkload()
399 FreeTensorIfUnused(m_InputToInputWeightsTensor); in FreeUnusedTensors()
H A DClLstmFloatWorkload.cpp67 m_InputToInputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload()
68 …BuildArmComputeTensor(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights->GetTensorInfo()); in ClLstmFloatWorkload()
82 lstm_param.set_cifg_params(m_InputToInputWeightsTensor.get(), in ClLstmFloatWorkload()
196 … InitializeArmComputeClTensorData(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights); in ClLstmFloatWorkload()
372 FreeTensorIfUnused(m_InputToInputWeightsTensor); in FreeUnusedTensors()
H A DClUnidirectionalSequenceLstmFloatWorkload.cpp183 m_InputToInputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClUnidirectionalSequenceLstmFloatWorkload()
184 …BuildArmComputeTensor(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights->GetTensorInfo()); in ClUnidirectionalSequenceLstmFloatWorkload()
198 lstm_param.set_cifg_params(m_InputToInputWeightsTensor.get(), in ClUnidirectionalSequenceLstmFloatWorkload()
363 … InitializeArmComputeClTensorData(*m_InputToInputWeightsTensor, m_Data.m_InputToInputWeights); in ClUnidirectionalSequenceLstmFloatWorkload()
879 FreeTensorIfUnused(m_InputToInputWeightsTensor); in FreeUnusedTensors()
H A DClQuantizedLstmWorkload.hpp35 std::unique_ptr<arm_compute::CLTensor> m_InputToInputWeightsTensor; member in armnn::ClQuantizedLstmWorkload
H A DClQLstmWorkload.hpp30 std::unique_ptr<arm_compute::CLTensor> m_InputToInputWeightsTensor; member in armnn::ClQLstmWorkload
H A DClLstmFloatWorkload.hpp34 std::unique_ptr<arm_compute::CLTensor> m_InputToInputWeightsTensor; member in armnn::ClLstmFloatWorkload
H A DClUnidirectionalSequenceLstmFloatWorkload.hpp47 std::unique_ptr<arm_compute::CLTensor> m_InputToInputWeightsTensor; member in armnn::ClUnidirectionalSequenceLstmFloatWorkload
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefLstmWorkload.cpp19 , m_InputToInputWeightsTensor (AssignScopedTensorHandle(descriptor.m_InputToInputWeights)) in RefLstmWorkload()
176 …m_InputToInputWeightsTensor->GetTensorInfo(), m_InputToInputWeightsTensor->GetConstTensor<void>()); in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.cpp23 , m_InputToInputWeightsTensor (AssignScopedTensorHandle(descriptor.m_InputToInputWeights)) in RefUnidirectionalSequenceLstmWorkload()
204 …m_InputToInputWeightsTensor->GetTensorInfo(), m_InputToInputWeightsTensor->GetConstTensor<void>()); in Execute()
H A DRefQLstmWorkload.cpp18 … , m_InputToInputWeightsTensor (AssignScopedTensorHandle(descriptor.m_InputToInputWeights)) in RefQLstmWorkload()
215 …m_InputToInputWeightsTensor->GetTensorInfo(), m_InputToInputWeightsTensor->GetConstTensor<void>()); in Execute()
H A DRefUnidirectionalSequenceLstmWorkload.hpp31 std::unique_ptr<ScopedTensorHandle> m_InputToInputWeightsTensor; member in armnn::RefUnidirectionalSequenceLstmWorkload
H A DRefLstmWorkload.hpp26 std::unique_ptr<ScopedTensorHandle> m_InputToInputWeightsTensor; member in armnn::RefLstmWorkload
H A DRefQLstmWorkload.hpp26 std::unique_ptr<ScopedTensorHandle> m_InputToInputWeightsTensor; member in armnn::RefQLstmWorkload