Home
last modified time | relevance | path

Searched defs:LstmQueueDescriptor (Results 1 – 3 of 3) sorted by relevance

/external/armnn/src/backends/backendsCommon/
DWorkloadData.hpp387 struct LstmQueueDescriptor : QueueDescriptorWithParameters<LstmDescriptor> struct
389 LstmQueueDescriptor() in LstmQueueDescriptor() function
414 const ConstCpuTensorHandle* m_InputToInputWeights;
415 const ConstCpuTensorHandle* m_InputToForgetWeights;
416 const ConstCpuTensorHandle* m_InputToCellWeights;
417 const ConstCpuTensorHandle* m_InputToOutputWeights;
418 const ConstCpuTensorHandle* m_RecurrentToInputWeights;
419 const ConstCpuTensorHandle* m_RecurrentToForgetWeights;
420 const ConstCpuTensorHandle* m_RecurrentToCellWeights;
421 const ConstCpuTensorHandle* m_RecurrentToOutputWeights;
[all …]
DWorkloadFactoryBase.hpp157 std::unique_ptr<IWorkload> CreateLstm(const LstmQueueDescriptor& /*descriptor*/, in CreateLstm() argument
DWorkloadFactory.cpp1477 std::unique_ptr<IWorkload> IWorkloadFactory::CreateLstm(const LstmQueueDescriptor& /*descriptor*/, in CreateLstm() argument