Home
last modified time | relevance | path

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

/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonUnidirectionalSequenceLstmWorkload.cpp54 TensorShape inputLayerShape = static_cast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetShape(); in NeonUnidirectionalSequenceLstmWorkload() local
500 TensorShape inputLayerShape = input.GetShape(); in NeonUnidirectionalSequenceLstmWorkloadValidate() local
H A DNeonUnidirectionalSequenceLstmFloatWorkload.cpp50 TensorShape inputLayerShape = static_cast<IAclTensorHandle*>(m_Data.m_Inputs[0])->GetShape(); in NeonUnidirectionalSequenceLstmFloatWorkload() local
519 TensorShape inputLayerShape = input.GetShape(); in NeonUnidirectionalSequenceLstmFloatWorkloadValidate() local
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClUnidirectionalSequenceLstmFloatWorkload.cpp52 TensorShape inputLayerShape = static_cast<IClTensorHandle*>(m_Data.m_Inputs[0])->GetShape(); in ClUnidirectionalSequenceLstmFloatWorkload() local
519 TensorShape inputLayerShape = input.GetShape(); in ClUnidirectionalSequenceLstmFloatWorkloadValidate() local