Home
last modified time | relevance | path

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

/aosp_15_r20/external/android-nn-driver/
H A DConversionUtils_1_3.hpp633 const TensorInfo& cellStateOutInfo = GetTensorInfoForOperand(*cellStateOut); in ConvertQuantizedLstm() local
690 { in ConvertQuantizedLstm()
H A DConversionUtils_1_2.hpp1951 const TensorInfo& cellStateOutInfo = GetTensorInfoForOperand(*cellStateOut); in ConvertQuantized16BitLstm() local
2768 const TensorInfo& cellStateOutInfo = GetTensorInfoForOperand(*cellStateOut); in ConvertLstm() local
3440 armnn::TensorInfo cellStateOutInfo({batchSize, numUnits}, cellStateInInfo.GetDataType(), in ConvertUnidirectionalSequenceLstm() local
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefUnidirectionalSequenceLstmWorkload.cpp64 TensorInfo cellStateOutInfo = GetTensorInfo(outputs[1]); in Execute() local
H A DRefQLstmWorkload.cpp71 const TensorInfo& cellStateOutInfo = GetTensorInfo(outputs[1]); in Execute() local
/aosp_15_r20/external/android-nn-driver/1.0/
H A DHalPolicy.cpp427 const armnn::TensorInfo& cellStateOutInfo = GetTensorInfoForOperand(*cellStateOut); in ConvertLstm() local
/aosp_15_r20/external/armnn/shim/sl/canonical/
H A DConverter.cpp3239 const TensorInfo& cellStateOutInfo = GetTensorInfoForOperand(*cellStateOut); in ConvertLstm() local
4343 const TensorInfo& cellStateOutInfo = GetTensorInfoForOperand(*cellStateOut); in ConvertQuantizedLstm() local
4400 { in ConvertQuantizedLstm()
4605 const TensorInfo& cellStateOutInfo = GetTensorInfoForOperand(*cellStateOut); in ConvertQuantized16BitLstm() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadData.cpp3066 auto cellStateOutInfo = workloadInfo.m_OutputTensorInfos[1]; in Validate() local
3351 auto cellStateOutInfo = workloadInfo.m_OutputTensorInfos[0]; in Validate() local
/aosp_15_r20/external/armnn/src/armnnDeserializer/
H A DDeserializer.cpp3479 armnn::TensorInfo cellStateOutInfo = ToTensorInfo(outputs[1]); in ParseQLstm() local