Home
last modified time | relevance | path

Searched defs:cellStateInInfo (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/armnn/delegate/classic/src/
H A DLstm.hpp113 …armnn::TensorInfo cellStateInInfo = GetTensorInfoForTfLiteTensor(tfLiteTensors[tfLiteNode->inputs-… in VisitLstmOperator() local
H A DUnidirectionalSequenceLstm.hpp115 …armnn::TensorInfo cellStateInInfo = GetTensorInfoForTfLiteTensor(tfLiteTensors[tfLiteNode->inputs-… in VisitUnidirectionalSequenceLstmOperator() local
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefQLstmWorkload.cpp68 const TensorInfo& cellStateInInfo = GetTensorInfo(inputs[2]); in Execute() local
/aosp_15_r20/external/android-nn-driver/1.0/
H A DHalPolicy.cpp422 const armnn::TensorInfo& cellStateInInfo = cellStateIn.GetTensorInfo(); in ConvertLstm() local
/aosp_15_r20/external/android-nn-driver/
H A DConversionUtils_1_2.hpp2763 const TensorInfo& cellStateInInfo = cellStateIn.GetTensorInfo(); in ConvertLstm() local
3427 const TensorInfo& cellStateInInfo = cellStateIn.GetTensorInfo(); in ConvertUnidirectionalSequenceLstm() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadData.cpp3063 auto cellStateInInfo = workloadInfo.m_InputTensorInfos[2]; in Validate() local
3348 auto cellStateInInfo = workloadInfo.m_InputTensorInfos[1]; in Validate() local
/aosp_15_r20/external/armnn/shim/sl/canonical/
H A DConverter.cpp3234 const TensorInfo& cellStateInInfo = cellStateIn.GetTensorInfo(); in ConvertLstm() local
/aosp_15_r20/external/armnn/src/armnnTfLiteParser/
H A DTfLiteParser.cpp3631 …armnn::TensorInfo cellStateInInfo = ToTensorInfo(subgraphPtr->tensors[operatorPtr->inputs[19]].get… in ParseUnidirectionalSequenceLSTM() local