Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonUnidirectionalSequenceLstmWorkload.cpp55 TensorShape outputLayerShape = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[2])->GetShape(); in NeonUnidirectionalSequenceLstmWorkload() local
501 TensorShape outputLayerShape = output.GetShape(); in NeonUnidirectionalSequenceLstmWorkloadValidate() local
H A DNeonUnidirectionalSequenceLstmFloatWorkload.cpp52 TensorShape outputLayerShape = static_cast<IAclTensorHandle*>(m_Data.m_Outputs[2])->GetShape(); in NeonUnidirectionalSequenceLstmFloatWorkload() local
520 TensorShape outputLayerShape = outputStateIn.GetShape(); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate() local
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClUnidirectionalSequenceLstmFloatWorkload.cpp54 TensorShape outputLayerShape = static_cast<IClTensorHandle*>(m_Data.m_Outputs[2])->GetShape(); in ClUnidirectionalSequenceLstmFloatWorkload() local
520 TensorShape outputLayerShape = outputStateIn.GetShape(); in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local