Home
last modified time | relevance | path

Searched refs:GetConstTensorForTfLiteTensor (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/armnn/delegate/classic/src/
H A DLstm.hpp57 params.m_InputToInputWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 1); in VisitLstmOperator()
60 params.m_InputToForgetWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 2); in VisitLstmOperator()
61 params.m_InputToCellWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 3); in VisitLstmOperator()
62 params.m_InputToOutputWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 4); in VisitLstmOperator()
67 … params.m_RecurrentToInputWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 5); in VisitLstmOperator()
70 params.m_RecurrentToForgetWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 6); in VisitLstmOperator()
71 params.m_RecurrentToCellWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 7); in VisitLstmOperator()
72 params.m_RecurrentToOutputWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 8); in VisitLstmOperator()
77 params.m_CellToInputWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 9); in VisitLstmOperator()
82 params.m_CellToForgetWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 10); in VisitLstmOperator()
[all …]
H A DUnidirectionalSequenceLstm.hpp59 params.m_InputToInputWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 1); in VisitUnidirectionalSequenceLstmOperator()
62 params.m_InputToForgetWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 2); in VisitUnidirectionalSequenceLstmOperator()
63 params.m_InputToCellWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 3); in VisitUnidirectionalSequenceLstmOperator()
64 params.m_InputToOutputWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 4); in VisitUnidirectionalSequenceLstmOperator()
69 … params.m_RecurrentToInputWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 5); in VisitUnidirectionalSequenceLstmOperator()
72 params.m_RecurrentToForgetWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 6); in VisitUnidirectionalSequenceLstmOperator()
73 params.m_RecurrentToCellWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 7); in VisitUnidirectionalSequenceLstmOperator()
74 params.m_RecurrentToOutputWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 8); in VisitUnidirectionalSequenceLstmOperator()
79 params.m_CellToInputWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 9); in VisitUnidirectionalSequenceLstmOperator()
84 params.m_CellToForgetWeights = GetConstTensorForTfLiteTensor(tfLiteTensors, tfLiteNode, 10); in VisitUnidirectionalSequenceLstmOperator()
[all …]
H A DClassicDelegateUtils.hpp460 armnn::ConstTensor* GetConstTensorForTfLiteTensor(const TfLiteTensor* tfLiteTensors, TfLiteNode* tf… in GetConstTensorForTfLiteTensor() function
/aosp_15_r20/external/armnn/delegate/opaque/src/
H A DOpaqueDelegateUtils.hpp540 armnn::ConstTensor* GetConstTensorForTfLiteTensor(const TfLiteOpaqueContext* tfLiteContext, in GetConstTensorForTfLiteTensor() function