Home
last modified time | relevance | path

Searched refs:m_RecurrentToCellWeightsTensor (Results 1 – 16 of 16) sorted by relevance

/external/armnn/src/backends/cl/workloads/
DClQuantizedLstmWorkload.cpp86 m_RecurrentToCellWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQuantizedLstmWorkload()
87 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in ClQuantizedLstmWorkload()
114 … m_RecurrentToCellWeightsTensor.get(), m_RecurrentToOutputWeightsTensor.get(), in ClQuantizedLstmWorkload()
125 …InitializeArmComputeClTensorData(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeight… in ClQuantizedLstmWorkload()
150 FreeTensorIfUnused(m_RecurrentToCellWeightsTensor); in FreeUnusedTensors()
DClQLstmWorkload.cpp35 m_RecurrentToCellWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload()
36 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in ClQLstmWorkload()
159 m_RecurrentToCellWeightsTensor.get(), in ClQLstmWorkload()
177 …InitializeArmComputeClTensorData(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeight… in ClQLstmWorkload()
393 FreeTensorIfUnused(m_RecurrentToCellWeightsTensor); in FreeUnusedTensors()
DClLstmFloatWorkload.cpp40 m_RecurrentToCellWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload()
41 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in ClLstmFloatWorkload()
190 … m_RecurrentToCellWeightsTensor.get(), m_RecurrentToOutputWeightsTensor.get(), in ClLstmFloatWorkload()
202 …InitializeArmComputeClTensorData(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeight… in ClLstmFloatWorkload()
419 FreeTensorIfUnused(m_RecurrentToCellWeightsTensor); in FreeUnusedTensors()
DClQuantizedLstmWorkload.hpp37 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToCellWeightsTensor; member in armnn::ClQuantizedLstmWorkload
DClQLstmWorkload.hpp35 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToCellWeightsTensor; member in armnn::ClQLstmWorkload
DClLstmFloatWorkload.hpp33 std::unique_ptr<arm_compute::CLTensor> m_RecurrentToCellWeightsTensor; member in armnn::ClLstmFloatWorkload
/external/armnn/src/backends/neon/workloads/
DNeonQuantizedLstmWorkload.cpp40 m_RecurrentToCellWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQuantizedLstmWorkload()
41 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in NeonQuantizedLstmWorkload()
72 m_RecurrentToCellWeightsTensor.get(), in NeonQuantizedLstmWorkload()
101 InitializeArmComputeTensorData(*m_RecurrentToCellWeightsTensor, in NeonQuantizedLstmWorkload()
199 FreeTensorIfUnused(m_RecurrentToCellWeightsTensor); in FreeUnusedTensors()
DNeonQLstmWorkload.cpp35 m_RecurrentToCellWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload()
36 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in NeonQLstmWorkload()
159 m_RecurrentToCellWeightsTensor.get(), in NeonQLstmWorkload()
177 …InitializeArmComputeTensorData(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights); in NeonQLstmWorkload()
395 FreeTensorIfUnused(m_RecurrentToCellWeightsTensor); in FreeUnusedTensors()
DNeonLstmFloatWorkload.cpp37 m_RecurrentToCellWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload()
38 …BuildArmComputeTensor(*m_RecurrentToCellWeightsTensor, m_Data.m_RecurrentToCellWeights->GetTensorI… in NeonLstmFloatWorkload()
190 … m_RecurrentToCellWeightsTensor.get(), m_RecurrentToOutputWeightsTensor.get(), in NeonLstmFloatWorkload()
206 InitializeArmComputeTensorData(*m_RecurrentToCellWeightsTensor, in NeonLstmFloatWorkload()
442 FreeTensorIfUnused(m_RecurrentToCellWeightsTensor); in FreeUnusedTensors()
DNeonQuantizedLstmWorkload.hpp34 std::unique_ptr<arm_compute::Tensor> m_RecurrentToCellWeightsTensor; member in armnn::NeonQuantizedLstmWorkload
DNeonQLstmWorkload.hpp35 std::unique_ptr<arm_compute::Tensor> m_RecurrentToCellWeightsTensor; member in armnn::NeonQLstmWorkload
DNeonLstmFloatWorkload.hpp34 std::unique_ptr<arm_compute::Tensor> m_RecurrentToCellWeightsTensor; member in armnn::NeonLstmFloatWorkload
/external/armnn/src/backends/reference/workloads/
DRefQLstmWorkload.hpp31 std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToCellWeightsTensor; member in armnn::RefQLstmWorkload
DRefLstmWorkload.hpp30 std::unique_ptr<ScopedCpuTensorHandle> m_RecurrentToCellWeightsTensor; member in armnn::RefLstmWorkload
DRefLstmWorkload.cpp24 …, m_RecurrentToCellWeightsTensor (AssignScopedCpuTensorHandle(descriptor.m_RecurrentToCellWeights… in RefLstmWorkload()
123 …m_RecurrentToCellWeightsTensor->GetTensorInfo(), m_RecurrentToCellWeightsTensor->GetTensor<void>()… in Execute()
DRefQLstmWorkload.cpp25 …, m_RecurrentToCellWeightsTensor (AssignScopedCpuTensorHandle(descriptor.m_RecurrentToCellWeights… in RefQLstmWorkload()
113 …m_RecurrentToCellWeightsTensor->GetTensorInfo(), m_RecurrentToCellWeightsTensor->GetTensor<void>()… in Execute()