Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/include/armnn/backends/
H A DWorkloadData.hpp395 struct LstmQueueDescriptor : QueueDescriptorWithParameters<LstmDescriptor> struct
397 LstmQueueDescriptor() in LstmQueueDescriptor() function
422 const ConstTensorHandle* m_InputToInputWeights;
423 const ConstTensorHandle* m_InputToForgetWeights;
424 const ConstTensorHandle* m_InputToCellWeights;
425 const ConstTensorHandle* m_InputToOutputWeights;
426 const ConstTensorHandle* m_RecurrentToInputWeights;
427 const ConstTensorHandle* m_RecurrentToForgetWeights;
428 const ConstTensorHandle* m_RecurrentToCellWeights;
429 const ConstTensorHandle* m_RecurrentToOutputWeights;
[all …]
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadFactoryBase.hpp151 std::unique_ptr<IWorkload> CreateLstm(const LstmQueueDescriptor& /*descriptor*/, in CreateLstm() argument
H A DWorkloadFactory.cpp2131 std::unique_ptr<IWorkload> IWorkloadFactory::CreateLstm(const LstmQueueDescriptor& /*descriptor*/, in CreateLstm() argument