Home
last modified time | relevance | path

Searched refs:m_CellToInputWeightsTensor (Results 1 – 12 of 12) sorted by relevance

/external/armnn/src/backends/neon/workloads/
DNeonQLstmWorkload.cpp53 m_CellToInputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonQLstmWorkload()
58 … BuildArmComputeTensor(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights->GetTensorInfo()); in NeonQLstmWorkload()
130 m_Data.m_CellToInputWeights != nullptr ? m_CellToInputWeightsTensor.get() : nullptr, in NeonQLstmWorkload()
206 … InitializeArmComputeTensorData(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights); in NeonQLstmWorkload()
398 FreeTensorIfUnused(m_CellToInputWeightsTensor); in FreeUnusedTensors()
DNeonLstmFloatWorkload.cpp61 m_CellToInputWeightsTensor = std::make_unique<arm_compute::Tensor>(); in NeonLstmFloatWorkload()
64 … BuildArmComputeTensor(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights->GetTensorInfo()); in NeonLstmFloatWorkload()
72 … m_Data.m_CellToInputWeights != nullptr ? m_CellToInputWeightsTensor.get() : nullptr, in NeonLstmFloatWorkload()
225 InitializeArmComputeTensorData(*m_CellToInputWeightsTensor, in NeonLstmFloatWorkload()
444 FreeTensorIfUnused(m_CellToInputWeightsTensor); in FreeUnusedTensors()
DNeonQLstmWorkload.hpp38 std::unique_ptr<arm_compute::Tensor> m_CellToInputWeightsTensor; member in armnn::NeonQLstmWorkload
DNeonLstmFloatWorkload.hpp36 std::unique_ptr<arm_compute::Tensor> m_CellToInputWeightsTensor; member in armnn::NeonLstmFloatWorkload
/external/armnn/src/backends/cl/workloads/
DClQLstmWorkload.cpp53 m_CellToInputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClQLstmWorkload()
58 … BuildArmComputeTensor(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights->GetTensorInfo()); in ClQLstmWorkload()
130 m_Data.m_CellToInputWeights != nullptr ? m_CellToInputWeightsTensor.get() : nullptr, in ClQLstmWorkload()
206 … InitializeArmComputeClTensorData(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights); in ClQLstmWorkload()
396 FreeTensorIfUnused(m_CellToInputWeightsTensor); in FreeUnusedTensors()
DClLstmFloatWorkload.cpp64 m_CellToInputWeightsTensor = std::make_unique<arm_compute::CLTensor>(); in ClLstmFloatWorkload()
67 … BuildArmComputeTensor(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights->GetTensorInfo()); in ClLstmFloatWorkload()
75 … m_Data.m_CellToInputWeights != nullptr ? m_CellToInputWeightsTensor.get() : nullptr, in ClLstmFloatWorkload()
214 … InitializeArmComputeClTensorData(*m_CellToInputWeightsTensor, m_Data.m_CellToInputWeights); in ClLstmFloatWorkload()
421 FreeTensorIfUnused(m_CellToInputWeightsTensor); in FreeUnusedTensors()
DClQLstmWorkload.hpp38 std::unique_ptr<arm_compute::CLTensor> m_CellToInputWeightsTensor; member in armnn::ClQLstmWorkload
DClLstmFloatWorkload.hpp35 std::unique_ptr<arm_compute::CLTensor> m_CellToInputWeightsTensor; member in armnn::ClLstmFloatWorkload
/external/armnn/src/backends/reference/workloads/
DRefQLstmWorkload.hpp34 std::unique_ptr<ScopedCpuTensorHandle> m_CellToInputWeightsTensor; member in armnn::RefQLstmWorkload
DRefLstmWorkload.hpp32 std::unique_ptr<ScopedCpuTensorHandle> m_CellToInputWeightsTensor; member in armnn::RefLstmWorkload
DRefLstmWorkload.cpp26 , m_CellToInputWeightsTensor (AssignScopedCpuTensorHandle(descriptor.m_CellToInputWeights)) in RefLstmWorkload()
183 … m_CellToInputWeightsTensor->GetTensorInfo(), m_CellToInputWeightsTensor->GetTensor<void>()); in Execute()
DRefQLstmWorkload.cpp28 … , m_CellToInputWeightsTensor (AssignScopedCpuTensorHandle(descriptor.m_CellToInputWeights)) in RefQLstmWorkload()
211 … m_CellToInputWeightsTensor->GetTensorInfo(), m_CellToInputWeightsTensor->GetTensor<void>()); in Execute()